pytroch官方网站:https://pytorch.org/
-
安装pytorch命令示例:
pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
-
2022年05月:windows RTX3080安装命令:
电脑驱动版本为11.6,cuda=10.6 cudnn=10.x
pip install torch==1.10.1+cu113 torchvision=0.11.2+cu113 -f https://download.pytorch.org/whl/torch_stable.html
pytorch、torchvision,cu的对应关系
资料来源:https://codechina.csdn.net/mirrors/pytorch/vision?utm_source=csdn_github_accelerator
可参考资源网站
https://download.pytorch.org/whl/torch_stable.html
已成功安装:
- 版本
torch 1.4.0+cu92
torchvision 0.5.0+cu92