win10+Anaconda+pytorch GPU版本配置(清华源踩坑实录)

       **

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即可

测试一下
在这里插入图片描述
完成

推荐参考
参考1
参考2

第一次发博客,有点小问题请多多海涵
设备情况不同,仅供参考哈

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值