参考博客:https://huaweicloud.youkuaiyun.com/637f7c6adacf622b8df85eb6.html?spm=1001.2101.3001.6650.1&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7Eactivity-1-124681233-blog-124571570.pc_relevant_vip_default&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7Eactivity-1-124681233-blog-124571570.pc_relevant_vip_default&utm_relevant_index=2
1.在Anaconda下建立pytorch虚拟环境。
conda create -n pytorch python=3.8
2.在该链接下载对应的torch和torchvision:
https://download.pytorch.org/whl/torch_stable.html
这几个都要对应好,cu113就是cuda11.3,绿色箭头是torch版本,蓝色39就是python版本3.9,然后都有linux和windows两种,选择Windows版本,根据你自己的各个版本去下载对应的whl文件。
在该连接下查看torch和python版本对应的torchvision版本:
https://github.com/pytorch/vision#installation
按照torch和torchvision那张图去找你对应版本,比如torch1.5.0对应的torchvision版本就是0.6.0。
3.下载好两个whl文件后,用anaconda Prompt切换到你的虚拟环境,就是上文那个命令行,在你的虚拟环境中切换到你两个whl文件的安装目录(cd…是倒退,cd+文件名是进入)。
在文件名前加pip install去安装这两个文件。
4.安装完成后就好了,再加以验证: