win10使用清华源快速安装pytorch-GPU版(推荐)
检查自己的cuda是否安装好
在anacondaprompt中输入
nvcc-V
配置清华园下载环境
同样在在anacondaprompt中输入
condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/ condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ condaconfig--setshow_channel_urlsyes
然后删除:C:\Users\Administrator目录下的.condarc文件,打开文件删除defaults那一行。如下图所示:
点击进入pytorch官网
官网链接: https://pytorch.org/get-started/locally/.
自己选择自己的版本,如下:
配置自己cuda版本的pytorch,cuda版本过低怎么办
如上所说,直接将cudatoolkit=10.1改为你自己的cuda版本,如我的是cuda9.0:
condainstallpytorchtorchvisioncudatoolkit=9.0pytorch
安装就行了。
安装指定版本的torchvision
如果你在看代码的时候,有些时候作者要求指定版本的torch或者torchvision,这个只能在官网上下载以前版本的文件。
验证你的pytorch安装好没有
python importtorch exit()
恭喜你安装成功。
总结
以上所述是小编给大家介绍的win10使用清华源快速安装pytorch-GPU版,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对毛票票网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:czq8825#qq.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。