神剑山庄资源网 Design By www.hcban.com

检查自己的cuda是否安装好

在anaconda prompt中输入

nvcc -V

win10使用清华源快速安装pytorch-GPU版(推荐)

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

配置清华园下载环境

同样在在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

然后删除:C:\Users\Administrator 目录下的 .condarc 文件,打开文件删除defaults那一行。如下图所示:


win10使用清华源快速安装pytorch-GPU版(推荐)

点击进入pytorch官网

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

自己选择自己的版本,如下:


win10使用清华源快速安装pytorch-GPU版(推荐)

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

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

配置自己cuda版本的pytorch,cuda版本过低怎么办

如上所说,直接将cudatoolkit=10.1改为你自己的cuda版本,如我的是cuda 9.0:


win10使用清华源快速安装pytorch-GPU版(推荐)

conda install pytorch torchvision cudatoolkit=9.0 pytorch

安装就行了。

安装指定版本的torchvision

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


win10使用清华源快速安装pytorch-GPU版(推荐)

点击 Previous PyTorch Versions


win10使用清华源快速安装pytorch-GPU版(推荐)

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

验证你的pytorch安装好没有

win10使用清华源快速安装pytorch-GPU版(推荐)

这样就安装好了。

python
import torch
exit()

恭喜你安装成功。

总结

以上所述是小编给大家介绍的win10使用清华源快速安装pytorch-GPU版,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!

标签:
win10,安装pytorch-GPU,win10,pytorch-GPU

神剑山庄资源网 Design By www.hcban.com
神剑山庄资源网 免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
神剑山庄资源网 Design By www.hcban.com

评论“win10使用清华源快速安装pytorch-GPU版(推荐)”

暂无win10使用清华源快速安装pytorch-GPU版(推荐)的评论...

P70系列延期,华为新旗舰将在下月发布

3月20日消息,近期博主@数码闲聊站 透露,原定三月份发布的华为新旗舰P70系列延期发布,预计4月份上市。

而博主@定焦数码 爆料,华为的P70系列在定位上已经超过了Mate60,成为了重要的旗舰系列之一。它肩负着重返影像领域顶尖的使命。那么这次P70会带来哪些令人惊艳的创新呢?

根据目前爆料的消息来看,华为P70系列将推出三个版本,其中P70和P70 Pro采用了三角形的摄像头模组设计,而P70 Art则采用了与上一代P60 Art相似的不规则形状设计。这样的外观是否好看见仁见智,但辨识度绝对拉满。