1、VS2015
软件正常方式安装,不多赘述。
2、CUDA以及cuDNN
CUDA 的下载地址:https://developer.nvidia.com/cuda-downloads
如果需要下载 CUDA 的历史版本,请移步:https://developer.nvidia.com/cuda-toolkit-archive
cuDNN的下载地址:https://developer.nvidia.com/cudnn
下载之前需要注册,然后问一系列问题,请耐心弄完。然后就可以下载了。
注:若使用其他版本的CUDA需要下载对应的cuDNN
对应版本查看网址:https://developer.nvidia.com/rdp/cudnn-archive
2.1安装CUDA8.0(PS:安装CUDA之前需要确定安装了显卡驱动程序)
安装CUDA时,一般双击安装文件,即可一直next完成安装。
2.2 cuDNN安装
将cuDNN解压后得到三个文件夹:
分别将这三个文件夹下的内容替换CUDA安装目录下对应文件夹下的内容,如我的是“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1”。
通过以上即可完成CUDA以及cuDNN的安装。
3 安装pytorch-gpu
在anaconda中新建pytorch虚拟环境,然后单击左键弹出一个菜单,选择第一个open teminal,进入窗口命令程序。
输入命令:pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl
安装 torchvision
输入命令:pip install torchvision
4 测试是否安装成功
在pytorch虚拟环境,然后单击左键弹出一个菜单,选择第一个open teminal,进入窗口命令程序。
进入Python环境,输入命令:import torch
如果没有报错,则说明安装成功