win10使用清华源快速安装pytorch-GPU版(推荐)
作者:来自星星的菜鸟 时间:2021-05-01 03:18:57
检查自己的cuda是否安装好
在anaconda prompt中输入
nvcc -V
显示如上面表示安装好了。
配置清华园下载环境
同样在在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那一行。如下图所示:
点击进入pytorch官网
官网链接: https://pytorch.org/get-started/locally/.
自己选择自己的版本,如下:
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:
conda install pytorch torchvision cudatoolkit=9.0 pytorch
安装就行了。
安装指定版本的torchvision
如果你在看代码的时候,有些时候作者要求指定版本的torch或者torchvision,这个只能在官网上下载以前版本的文件。
点击 Previous PyTorch Versions
上面就有你想要找到的版本对应文件,下载下来 pip install
就好了。需要注意很多版本问题,如:cuda9.0的torchvison只支持到0.3.0版本等等。
验证你的pytorch安装好没有
这样就安装好了。
python
import torch
exit()
恭喜你安装成功。
总结
以上所述是小编给大家介绍的win10使用清华源快速安装pytorch-GPU版,网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!
来源:https://blog.csdn.net/weixin_42954710/article/details/102702243


猜你喜欢
我的四年做地方站经历!论坛运营篇
Exchange 2003安全:用筛选器反垃圾邮件

VMware虚拟机三种网络模式区别

电信屏蔽Godaddy部分DNS服务
关于baidu一些排名规则的讨论
详解linux centos yum安装LAMP环境
建站先磨刀 网站整体策划注意的几个要点
简易性可用性:网站登录界面优化八点建议
巧用美图秀秀玩出万圣节新花样

简明教程:Google AdSense 投放政策
Docker部署nGrinder性能测试平台过程解析

如何开通Godaddy免费空间做图片外链空间

适合新手做的网赚项目总结
GoDaddy:如何下载Account Manager的文件
Win10上VMware workstation安装图文教程

轻松解决域名DNS解析的故障
详解Linux误删用户家目录的恢复方法
网游企业尝试按日消费 金山推包日2.5元模式
风讯FoosunCMS 5.0免费版正式发布

从小切口看引擎对原创文章的抓取与排名
