pip安装pytorch清华_win10使用清华源快速安装pytorchGPU版(推荐)

检查自己的cuda是否安装好

在anaconda prompt中输入

nvcc -V

win10使用清华源快速安装pytorch-GPU版(推荐)-1.jpg (14.94 KB, 下载次数: 0)

2019-11-28 17:15 上传

显示如上面表示安装好了。

配置清华园下载环境

同样在在anaconda prompt中输入

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --set show_channel_urls yes然后删除&＃xff1a;C:\Users\Administrator 目录下的 .condarc 文件&＃xff0c;打开文件删除defaults那一行。如下图所示&＃xff1a;

win10使用清华源快速安装pytorch-GPU版(推荐)-2.jpg (44.92 KB, 下载次数: 0)

2019-11-28 17:15 上传

点击进入pytorch官网

官网链接:  https://pytorch.org/get-started/locally/.

自己选择自己的版本&＃xff0c;如下&＃xff1a;

win10使用清华源快速安装pytorch-GPU版(推荐)-3.jpg (41.67 KB, 下载次数: 0)

2019-11-28 17:15 上传

conda install pytorch torchvision cudatoolkit&＃61;10.1 -c pytorch

这行代码的含义是&＃xff1a;安装pytorch时会把torchvision也给安装上&＃xff0c;就不用后面再安装torchvision了。cudatoolkit&＃61;10.1表示cuda版本10.1。-c表示使用官网下载。不含-cpu表示gpu版。

配置自己cuda版本的pytorch&＃xff0c;cuda版本过低怎么办

如上所说&＃xff0c;直接将cudatoolkit&＃61;10.1改为你自己的cuda版本&＃xff0c;如我的是cuda 9.0&＃xff1a;

win10使用清华源快速安装pytorch-GPU版(推荐)-4.jpg (21 KB, 下载次数: 0)

2019-11-28 17:15 上传

conda install pytorch torchvision cudatoolkit&＃61;9.0 pytorch安装就行了。

安装指定版本的torchvision

如果你在看代码的时候&＃xff0c;有些时候作者要求指定版本的torch或者torchvision&＃xff0c;这个只能在官网上下载以前版本的文件。

win10使用清华源快速安装pytorch-GPU版(推荐)-5.jpg (17.36 KB, 下载次数: 0)

2019-11-28 17:15 上传

点击 Previous PyTorch Versions

win10使用清华源快速安装pytorch-GPU版(推荐)-6.jpg (32.67 KB, 下载次数: 0)

2019-11-28 17:15 上传

上面就有你想要找到的版本对应文件&＃xff0c;下载下来 pip install就好了。需要注意很多版本问题&＃xff0c;如&＃xff1a;cuda9.0的torchvison只支持到0.3.0版本等等。

验证你的pytorch安装好没有

win10使用清华源快速安装pytorch-GPU版(推荐)-7.jpg (13.56 KB, 下载次数: 0)

2019-11-28 17:15 上传

这样就安装好了。

python

import torch

exit()恭喜你安装成功。

总结

以上所述是小编给大家介绍的win10使用清华源快速安装pytorch-GPU版,希望对大家有所帮助&＃xff0c;如果大家有任何疑问请给我留言&＃xff0c;小编会及时回复大家的。在此也非常感谢大家对咔叽网单www.2nzz.com网站的支持&＃xff01;

如果你觉得本文对你有帮助&＃xff0c;欢迎转载&＃xff0c;烦请注明出处&＃xff0c;谢谢&＃xff01;