安装pytorch最麻烦的就是版本对应
CUDA版本:10.2
Anaconda版本: 22.9.0,自带python 3.9.13
Pytorch torchvision,torchaudio 版本:
直接去官网找:
Previous PyTorch Versions | PyTorch
找这行命令:
CUDA 10.2的,这几个和python 3.9。13可以搭配使用
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu102/torch_stable.html
最后检查下:
ok,可以用了