Windows安装Pytorch并配置Anaconda与Pycharm

1 开发环境准备

Python 3.7+Anaconda3 5.3.1（64位）+CUDA+Pycharm Community

2 安装Anaconda

2.1 进入官网下载：

根据windows版本选择64位或32位进行下载，这里选择3.7版本64位进行下载。

2.2 下载后获得.exe文件，双击运行

2.2.1  点击“Next”

2.2.2  点击“I Agree”

2.2.3  点击“Next”

2.2.4  默认安装路径，点击“Next”

2.2.5  注意勾选“Add Anaconda3 to my PATH environment variable”，再点击“Install”

2.2.6  等待安装完成（时间约为3~5分钟）

2.2.7  点击“Next”

2.2.8  点击“Next”

2.2.9  点击“Finish”，安装完成

2.3 测试是否安装成功

按住键盘上的“win+R”进入windows运行，输入“cmd”进入命令行窗口，输入命令“conda list”。如果出现以下很多库文件名称，则表示Anaconda已经安装成功。

3 安装CUDA

3.1 查看支持CUDA的GPU型号

安装CUDA的必要条件之一就是要求电脑拥有Nvidia显卡，同时需要提前查看支持CUDA的GPU型号。

3.2 进入官网下载

这里选择目标平台的配置为“Windows+x84_64+10+exe[local]”，参照下图进行勾选，点击“Download”下载文件到本地。

3.3 下载后获得.exe文件，双击运行

3.3.1  点击“OK”

选择默认路径，点击“OK”进行安装。

3.3.2  点击“Next”

等待约10分钟，出现以下界面，点击“同意并继续”。

3.3.3  点击“下一步”

选择“精简”，点击“下一步”。

3.3.4  点击“NEXT”

勾选“I understand，……”，点击“NEXT”。安装完成后，界面自动消失。

3.4 确定安装成功

3.4.1  安装确认

找到安装的文件夹，这里默认是“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin”，查看是否存在“nvcc.exe”文件。

3.4.2  安装测试

按住键盘上的“win+R”进入windows运行，输入“cmd”进入命令行窗口，输入命令“nvcc -V”。如果出现下图版本信息，则表示CUDA已经安装成功。

4 安装Pytorch

4.1 进入官网

进入官网，选择首选项为“Stable（1.4）+Windows+Conda+Python+10.1”，即可获得安装的命令“conda install pytorch torchvision cudatoolkit=10.1 -c pytorch”。

4.2 下载后获得.exe文件，双击运行

4.2.1  以管理员身份打开命令提示符

点击“开始”，找到“Windows系统”里的“命令提示符”，选择“以管理员身份运行”即可打开对话框。

4.2.2  将4步骤4.1所得到的命令复制到命令提示符界面中（不建议使用这种方法）

由于国内网络的限制，不建议直接使用此方法进行下载安装，可直接跳过步骤4.2.2直接转到步骤4.2.3进行操作。

等待电脑环境自动检测完成

在弹出“Proceed（[y]/n）?”后，输入“y”并回车即可进行下载安装。

4.2.3  更换国内源（如清华镜像源）进行下载安装

在已经打开的命令行中输入以下代码：

1 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
2
3 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
4
5 conda config --set show_channel_urls yes
6
7 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
8

然后将Pytorch官网上获取的代码复制到命令行中，并删除后面的“-c pytorch”，回车即可进行环境的自动检测。

1 conda install pytorch torchvision cudatoolkit=10.1

在弹出“Proceed（[y]/n）?”后，输入“y”并回车即可进行下载安装。

过一段时间后，安装自动完成，界面如下图所示。

5 安装Pycharm

5.1 进入官网

进入官网，选择右侧免费的社区版（Community）进行下载。

5.2 下载后获得.exe文件，双击运行

5.2.1 点击“Next”

5.2.2 默认安装路径，点击“Next”

5.2.3 选择安装选项，点击“Next”

勾选“64-bit launcher”、“.py”后，点击“Next”。

5.2.4 点击“Install”

5.2.5 等待安装

5.2.6 点击“Finish”，安装完成

5.2  编译环境配置

5.3.1 打开Pycharm，点击“Create New Project”

5.3.2 默认工程名，点击“Create”

5.3.3点击“File”→“Create”→“Settings...”

5.3.4点击“Project Interpreter”

勾选“Project Interpreter”，点击设置按钮，选择“Show All…”。

5.3.5 点击“Project Interpreter”

勾选“Existing environment”，选择“C:\Users\Wang Qi\anaconda3\python.exe”，如未显示则按此路径寻找选中python.exe。勾选“Make available to all projects”，点击“OK”、“OK”保存退出即可。

5.3.6 新建main.py文件

选中当前工程文件夹“untitled”，右击选择“New”→“Python File”，随意命名一个文件即可，这里命名为“main”。

5.3.7 测试Pytorch是否安装成功

在“main.py”文件中输入以下代码并运行：

1 import torch
2
3 print(torch.__version__)
4 print('gpu:', torch.cuda.is_available())

如果输出如下图所示的内容，则表示Pytorch安装成功。

至此，Windows安装Pytorch并配置Anaconda与Pycharm所有工作都已完成，让我们一起开启Pytorch的学习使用之旅吧！

参考资源：

1、pytorch下载太慢的解决办法

2、手把手的教你安装PyCharm --Pycharm安装详细教程（一）（非常详细，非常实用）