**
win10+Anaconda+pytorch GPU版本配置(清华源踩坑实录)**
先查看显卡配置
win+r,运行cmd,nvidia-smi后回车
不需要更新驱动了
可以直接下载使用CUDA10的pytorch(pytorch官网没有CUDA11的对应版本)
默认先是配置好了Anaconda
Anaconda在安装时选择默认安装或者自定义都可以,里面有一个自动添加环境变量的操作,问题不大
在使用conda进行安装时遭遇了迎头痛击
同样是更换了清华源(如下)
打开anaconda prompt,输入下列指令:
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 create -n your_env_name python=X.X
your_env_name 是你自定义的虚拟环境名(比如用pytorch、pytorchGpu之类的,随你喜欢)
python版本号就如实填,我是python=3.6
激活环境 activate your_env_name
既然选择了用镜像源下载,那么安装指令的 -c pytorch需删去,也就是:conda install pytorch torchvision cudatoolkit=10.1
问题来了
划重点!!!!!
在这里我惊奇的发现虽然下载的速度快的吓人,明明我选了GPU版的pytorch下载,最后给了一个CPU版本???
在anaconda prompt输入python
测试
import torch
print(torch.cuda.is_available())
import torch倒是没问题
print(torch.cuda.is_available())返回了一个None
这下载的是CPU版本吧???
这里有参考
尝试了无数种方法以后,我在anaconda prompt选中版本号直接下载(还是用的清华源)
类似于
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch
结果成功了,速度还挺快
pycharm配置
打开File
点击settings
Interpreter选择打开你安装anaconda的位置,选择你自己虚拟环境下的python.exe即可
测试一下
完成
第一次发博客,有点小问题请多多海涵
设备情况不同,仅供参考哈