『PyTorch』安装教程

1. 安装Anaconda

• 下载地址：https://www.anaconda.com/，傻瓜式安装，记得勾选【add to path】
• 要学会创建虚拟环境，这个百度一大把

2. 安装pycharm

• 下载地址：https://www.jetbrains.com/pycharm/，基本也是傻瓜式安装，也要记得勾选【add to path】
• 这里可以安装社区版，专业版你要能找到破解方法破解当然更好

3. pytorch安装(GPU版本)

一般国内直接安装太慢，这里使用离线安装。

• python版本：3.6
• cuda版本：9.2
• pytorch版本：1.4.0
• torchvision版本：0.5.0

3.1 下载cuda安装包

• 点击cuda9.2进入下载页面

• 根据自己的系统选择
  

• 开始下载安装包
  

3.2 下载cudnn安装包

• 点击cudnn进入下载页面(此下载必须登录账号，没有就先注册)

• 开始下载
  

• 选择相对应cuda版本下载
  

• 选择对应操作系统的版本
  

3.3 安装cuda

• Step 1
  运行安装包，选择安装路径

• Step 2
  选择【精简】模式，点击【下一步】
  

• Step 3
  勾选【接受】，点击【NEXT】
  

• Step 4
  点击【下一步】
  

• Step 5
  取消勾选【创建桌面快捷方式】，点击【关闭】
  

• Step 6
  验证cuda是否安装成功
  按住Win+E，输入【cmd】，打开命令行，输入

  nvcc -V
  

  有如下返回则安装成功：
  

3.4 安装cudnn

• Step 1
  解压压缩包，压缩包内文件如下：
  

• Step 2
  打开cuda安装文件夹，默认的强制安装路径为：

                                         C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2

  之前选择的路径是解压路径，安装完成默认删除，打开安装路径如下：
  

• Step 3
  将解压的cudnn压缩包内的三个文件夹复制到cuda安装目录下：
  
  

• Step 4
  验证cudnn是否安装完成按住Win+E，输入【cmd】，打开命令行，输入

  cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\extras\demo_suite
  

  进入demo文件夹，然后执行两个执行程序：

  bandwidthTest.exe
  

  

  deviceQuery.exe
  

  

3.5 下载pytorch离线文件

• 进入pytorch官网：https://pytorch.org/，点击【Get Started】

  

• 选择适合电脑的配置

  

  记住勾选【pip】，这样才会出现whl文件的网址

• 打开whl文件网址
  因为直接运行此代码太慢了，可以复制其中的网址，打开后选择适合自己的版本下载

  

  网址打开后如下所示：

  

此处文件的命名时有规律的：

• 选择需要的版本下载，可以用Ctrl+F搜索，找到所需文件单击下载

  

  

3.6 安装pytorch

• 打开pycharm，新建project
  
  这里要选择【Existing interpreter】，选择你创建的anaconda虚拟环境，第一次创建的话，可能不会直接显示，你需要自己添加虚拟环境的解释器：
  
  
  

• 创建python file，写入以下内容

  import torch
  
  print(f"Hello PyTorch {torch.__version__}")
  

  运行后会显示以下错误：

  ModuleNotFoundError: No module named 'torch'`

  下面就开始安装所缺的包

• 打开pycharm的【Terminal】窗口
  

  输入 cd 命令，进入前面所下载的离线文件的路径：
  

• 安装torch

  输入pip install torch，按住Tab键，即可自动补全：
  
  回车进行安装
  

  下面类似方法安装torchvision：

  

• 再次运行代码，可以发现正常运行
  
  OK，至此安装完成.