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