anaconda下载对应gpu版本的torch

自己记录一下
https://download.pytorch.org/whl/torch_stable.html

找到前面是cu+cuda版本的才是gpu，cpu开头的是下载cpu版的。我使用cuda11.5，这里可以看到torch使用1.11.0。
所以在anaconda里面输：

pip install torch==1.11.0+cu115 torchvision==0.12.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html
# torch==1.11.0(和上面需要下载的版本对应)+cu115(表明需要下载的是对应cuda版本11.5的gpu版的) 
# torchvision==0.12.0+cu115(和上面同理) 
# -f https://download.pytorch.org/whl/torch_stable.html (每次都加上）

个人常用：

pip install torch==1.11.0+cu115 torchvision==0.12.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html
pip install torch==1.10.0+cu113 torchvision==0.11.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
#链接上面的cu113可以加可以不加