anacoda中配置pytorch

    • 创建环境

1.1 添加镜像

由于本人的踩坑,建议使用镜像来下载相关代码,配置镜像(如清华镜像)见如下博客

添加镜像

1.2 创建环境

在base下输入以下代码

conda create -n pytorch python=3.6

pytorch为我们自定义的环境名称,您可根据自己的想法随意写。

由于我是跟着b站视频来的,这里的python版本也选择3.6,至于其它版本,未仔细研究。

1.3 激活环境

conda activate 您的环境名称

激活后,就看见base变成了您新建的环境,说明激活成功(我理解的就是切换环境)

    • 下载pytorch

2.1 查看自己GPU的适配CUDA版本

由于我的电脑是有GPU的,因此 这里也只是针对这种情况

打开cmd,输入命令

nvidia-smi
anacoda中配置pytorch_第1张图片

如上,我电脑的CUDA Version是11.1,因此在选择的时候不能超过该版本

2.2 下载pytorch

首先去官网找到对应的pytorch:官网,看到如下

anacoda中配置pytorch_第2张图片

刚开始我选择的是conda下载,但是无论怎样(视频弹幕中的方法都尝试了),依旧下载不下来---一直在solving environment,最后我选择了pip下载。

但是我们发现,这里只有11.6和11.7,我最高使用11.1,怎么办呢?

点击下方的Previous....,从历史版本中选择

anacoda中配置pytorch_第3张图片

这里的10.2就是cuda版本,我用的也是这个,后来发现也是有11.1的版本,但是懒得换了

复制命令到新建的环境下即可

2.3 下载过程中的踩坑

命令执行后,下载了一段时间,出现error 28 no space for device,我上网搜,说是c盘空间不足,于是我就把c盘清理了好久,腾出了7个G,之后才成功,至于为什么,我也不清楚(doge)

    • 检查pytorch安装成功和能否使用GPU

输入

pip list

查看是否有torch

anacoda中配置pytorch_第4张图片

接下来查看python中能否使用torch,按下图中的指令,如果最后是true说明没有问题。

anacoda中配置pytorch_第5张图片

你可能感兴趣的:(pytorch,python)