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安装详细教程(一)(非常详细,非常实用)

posted @ 2020-04-08 02:51  wangsl97  阅读(1203)  评论(0编辑  收藏  举报