Pytorch环境安装【Python3.7+Anaconda3+CUDA10.1】

1、安装环境

OS:              Win7 旗舰版 x64
Python:      Python 3.7.7（通过Anaconda软件自带安装）
Anaconda: Anaconda3-2020.02-Windows-x86_64.exe
CUDA:        cuda_10.1.105_418.96_windows.exe  具体选择界面的下载地址 【我原先用的是10.2，后面有图片不一致的请原谅】

Pytorch：在Anaconda中，执行命令安装

 

 2、Anaconda安装过程

1.选择安装路径：C:\ProgramData\Anaconda3

2.勾选两个：选择添加环境变量，设置Anaconda3自带的Python3.7的为系统默认环境。

 

 

3.打开windows自带的命令窗口cmd.exe，测试安装是否成功

cmd输入conda list命令，输出结果如下：

 cmd输出命令：conda --version，结果输出版本号，代表安装成功。

3、CUDA安装过程

1.CUDA下载

百度搜索CUDA download，进入官方下载页面，选择适合的操作系统和install type:(local的exe)

 2、双击安装

检查通过的话，选中需要的Visual Studio 环境，进入安装环节

 3.检查cuda是否安装成功

先核实一下，目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin下，是否有nvcc.exe文件。

然后在cmd中输入指令： nvcc -V，可以看到我们版本号为Cuda 10.1

本质上，是我们系统环境变量里多了Cuda的路径配置和Anaconda的路径

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp;

C:\ProgramData\Anaconda3;C:\ProgramData\Anaconda3\Library\mingw-w64\bin;C:\ProgramData\Anaconda3\Library\usr\bin;C:\ProgramData\Anaconda3\Library\bin;C:\ProgramData\Anaconda3\Scripts;

4、安装Pytorch

1.首先进入Pytorch下载页面：https://pytorch.org/get-started/locally/

选择合适的版本10.1，然后拷贝命令：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

---

以管理员方式运行2.以管理员身份运行cmd.exe(开始菜单里搜cmd，然后右击以Administrator身份运行)

输入如下脚本命令：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

第一次由于网络原因下载失败。。。第二次才好使

3.选择y ，表示同意更新

 

4. 加速Pytorch的安装（手动添加国内镜像源）

1.打开Anaconda Navigator，点击Channels。

2.手动添加清华镜像源

https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

一个个将地址add进去，然后点击Update channels。

 

重新执行步骤4中的pytorch安装脚本，即可提速。（国内实在慢的，建议手动离线安装PKG文件，参考这篇文章《pytorch手动安装开发包》）

5. 验证pytorch是否安装成功

在开始菜单中，打开 Python控制台。

（1）输入 import torch，如果没有报错，意味着 PyTorch 已经顺利安装了

（2）接下来，输入 torch.cuda.is_available，如果是 True，意味着你可以使用 GPU，如果是 False，意味着只能使用CPU。