1、安装环境

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

Pytorch:Pytorch安装,建议参考我这篇离线手动安装,速度比较快。《Win10 + CUDA10.1+pytorch手动安装》,注意这篇,我后续用的都是win10系统了。

 2、Anaconda安装过程

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

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

pytorch支持python3.9吗 pytorch3.7_Pytorch安装

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

cmd输入conda list命令,输出结果如下:

pytorch支持python3.9吗 pytorch3.7_pytorch支持python3.9吗_02

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

pytorch支持python3.9吗 pytorch3.7_Pytorch安装_03

3、CUDA安装过程

1.CUDA下载

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

pytorch支持python3.9吗 pytorch3.7_pytorch支持python3.9吗_04

 2、双击安装

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

pytorch支持python3.9吗 pytorch3.7_pytorch_05

pytorch支持python3.9吗 pytorch3.7_pytorch_06

 3.检查cuda是否安装成功

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

pytorch支持python3.9吗 pytorch3.7_加速_07

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

pytorch支持python3.9吗 pytorch3.7_pytorch支持python3.9吗_08

本质上,是我们系统环境变量里多了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


pytorch支持python3.9吗 pytorch3.7_Pytorch安装_09

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

输入如下脚本命令:

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

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

pytorch支持python3.9吗 pytorch3.7_安装_10

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

pytorch支持python3.9吗 pytorch3.7_Pytorch安装_11

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

1.打开Anaconda Navigator,点击Channels。

pytorch支持python3.9吗 pytorch3.7_pytorch_12

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。

pytorch支持python3.9吗 pytorch3.7_pytorch_13

重新执行步骤4中的pytorch安装脚本,即可提速。

5. 验证pytorch是否安装成功

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

(1)输入 import torch,如果没有报错,意味着 PyTorch 已经顺利安装了

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

pytorch支持python3.9吗 pytorch3.7_Pytorch安装_14