1、安装环境

OS:              Win10 专业 x64
Python:      Python 3.7.7(通过Anaconda软件自带安装)
Anaconda: Anaconda3-2020.02-Windows-x86_64.exe 【这个版本自带的python是3.7.7】
CUDA:        cuda_10.1.105_418.96_windows.exe  ,根据操作系统不同,可以在线选择不同版本下载,界面网址为下载地址 【我原先用的是10.2,后面有图片不一致的请原谅】

Pytorch:官方是Pytorch在线指令安装,建议参考我后面的离线手动安装,速度比较快。

Pycharm:pycharm-community-2021.1.3.exe,开发工具,社区免费版,下载后,自行一路默认安装。

 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下载页面:Start Locally | PyTorch

选择合适的版本10.1,然后拷贝命令:

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch

更新conda版本

Please update conda by running

    $ conda update -n base -c defaults conda

2、手动下载安装包

下载地址:

Index of /anaconda/pkgs/main/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror(稳定,只需要下载conda文件)

Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror (最新,只能下载tar.bz2文件)

https://repo.anaconda.com/pkgs/main/win-64/(官网)

 具体下载安装包列表:

cudatoolkit        pkgs/main/win-64::cudatoolkit-10.1.243-h74a9793_0           下载:cudatoolkit-10.1.243-h74a9793_0.tar.bz2
  libuv              pkgs/main/win-64::libuv-1.40.0-he774522_0                            速度快,可以自动安装
  ninja              pkgs/main/win-64::ninja-1.10.1-py37h7ef1ec2_0                     速度快,可以自动安装
  pytorch            pytorch/win-64::pytorch-1.7.0-py3.7_cuda101_cudnn7_0     下载:pytorch-1.7.0-py3.7_cuda101_cudnn7_0.tar.bz2
  torchaudio         pytorch/win-64::torchaudio-0.7.0-py37                                 下载:torchaudio-0.7.0-py37.tar.bz2
  torchvision        pytorch/win-64::torchvision-0.8.1-py37_cu101                      下载:torchvision-0.8.1-py37_cu101.tar.bz2
  typing_extensions  pkgs/main/noarch::typing_extensions-3.7.4.3-py_0         速度快,可以自动安装

3.拷贝至指定路径下:C:\ProgramData\Anaconda3\pkgs

4.然后手动离线,按顺序安装下载包。

以管理身份打开cmd,输入cmd C:\ProgramData\Anaconda3\pkgs,进入安装包目录。然后按照顺序安装下载包

      conda install --use-local pytorch-1.7.0-py3.7_cuda101_cudnn7_0.tar.bz2

      conda install --use-local torchvision-0.8.1-py37_cu101.tar.bz2          

      conda install --use-local torchaudio-0.7.0-py37.tar.bz2

     conda install --use-local cudatoolkit-10.1.243-h74a9793_0.tar.bz2

安装成功结果:

5. 执行pytorch安装总指令,检查一遍是否完全安装

执行pytorch安装总指令,检查一遍是否完全安装。

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch (红色-c pytorch 去掉,免得从镜像上重新下载)

conda install pytorch torchvision torchaudio cudatoolkit=10.1

6. 验证pytorch是否安装成功

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

(1)输入 import torch,如果没有报错,意味着 PyTorch 已经顺利安装了(我的笔记本win10 home版本系统缺少VC_redist.64.exe,安装后也解决)

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


(换了幅图,原图代码torch.cuda.is_available(),忘加括号了)

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐