Windows10下快速安装PyTorch

刚刚成功装上了PyTorch,先说一下我的环境,windows10 + Pycharm + python 3.7 + Anaconda3

这里默认大家已经安装了Anaconda3+Pycharm+python 3.7,如果没有的话可以参考https://blog.csdn.net/CSS360/article/details/88215036

首先第一步,查看自己电脑的python版本,直接控制台进去输入python、

第二步,登录PyTorch官网,https://pytorch.org/ 在下面选择自己想要的版本,这是我的选择

第一行大多都是Stable,第二行选择自己的操作系统,第三行选择安装方式,因为大家都安装了Anaconda3,所以推荐Conda,第四行选择自己的python版本,第五行是你的CUDA版本,关于查看自己电脑是否支持CUDA,大家参考https://www.jianshu.com/p/d3b9419a0f89   我的版本是10.2所以我这里选择10.0

紧接着选择好之后 最下面的命令就是用来安装的

 

第三步,打开Anaconda的prompt

就是Anaconda Prompt,打开之后就是命令行的形式

第四步。由于我们使用conda形式进行安装,所以建议先更新一下conda

输入命令 conda upgrade conda 然后一路确认 可能会更新很多东西,等就完事儿了

第五步,更换镜像源,由于我们要下载的PyTorch的官方资源在国外,所以我们直接下载的话速度极慢而且会经常中断,原因众所周知,所以为了解决这个问题,我们把源修改为清华源

输入命令:

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 --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
第六步,把刚才在PyTorch官网的那条命令粘过来,这里有个坑,大家千万注意

原始命令后面有

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch 

一定要把  -c  pytorch删掉,因为这一句的作用是指定了conda获取pytorch的channel,在此指定为conda自带的pytorch仓库,不删掉这个的话 我们的更新还是很慢,并且会经常由于网络原因出错中断

 

第七步,我们在Pycharm中将py的解释器选择为Anaconda,这样我们就可以直接在pycharm中完成使用了

选择之后,通过 import torch进行导入,不报错就说明成功了!

 

上面的内容是我看了网上众多大佬的总结并且自己踩坑无数之后得出的方法,希望能帮到需要的人

(写代码前的配置环境过程总是令人头疼)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章