xxxxxxxxxx
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
xxxxxxxxxx
pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html
xxxxxxxxxx
from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)
xxxxxxxxxx
CUDA-10.2 PyTorch builds are no longer available for Windows, please use CUDA-11.3
pytorch cmd with cuda 11.6
xxxxxxxxxx
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia