使用Conda配置PyTorch

1. 添加channels

下载地址

$ 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/bioconda/
$ conda config --set show_channel_urls yes

2. 创建一个新环境

创建一个新python环境,为pytorch提供资源.

win10下找到 anaconda的提示行,输入如下:

(base) C:\Users\54328>conda create -n pytorch python=3.6

• -n: 表示 name 代表,后面接的环境的名称
• 注意: 一定国内一定要添加channels, 下载速度会变得很快~ 而且不容易被墙

等待一会儿…

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-678fKdz6-1588215073554)(C:\Users\54328\AppData\Roaming\Typora\typora-user-images\image-20200430003733154.png)]

安装完成~

conda env list

上面安装好了python 3.6的环境,并命名pytorch. 下面激活该环境

conda activate pytorch

当在小黑框里面看见(pytorch) C:\Users\54328的字样时证明安装成功。

下面查看当前环境下由哪些包

pip list

3. 安装pytorch

pytorch - 官网.

进入之后选择:

• PyTorch版本
• 操作系统
• 使用什么安装(windows下推荐conda)
• 语言(这里选择Python)
• CUDA: 这个是显卡相关的.
  • 查看GPU是否支持CUDA
  • 在上面如果找到了对应的型号: 那么勾选 CUDA 9.2
  • 如果没有找到,勾选 None

完成以上操作之后,你会得到一串安装PyTorch的指令.

之后你就打开Anaconda Prompt(win10aconda Prompt(win10在左下角的菜单里面找), 输入(你的也许和我不同):

conda install pytorch torchvision cudatoolkit=10.1

注意,官网会提供一个 -c pytorch意思是使用 PyTorch官网提供的channel. 这样在或内会下载很慢.因此 需要去掉

如果上面安装比较慢, 可以参考这篇文章.之后再尝试以上代码

4. 判断环境是否安装成果

[检验PyTorch是否安装成功]

• 命令行输入 python, 进入python环境.
• 输入import torch, 未报错则安装成功

[下面检验是否能使用GPU]

• 命令行输入torch.cuda.is_available(), 返回True则代表可以使用GPU