Win10环境下安装Pytorch GPU版【torch1.1.0+CUDA9.0+CUDNN7.1】

Win10环境下安装Pytorch GPU版【torch1.1.0+CUDA9.0+CUDNN7.1】

 本文介绍在Anaconda、CUDA、CUDNN都已安装完的基础上安装Pytorch GPU版的方法

（1）登录Pytorch官网：https://pytorch.org/

（2）创建虚拟环境

• 打开Anaconda Prompt

• 输入：conda create -n torch
  

• 激活环境：activate torch
  

• 输入 pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl

• 输入 pip install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

（3）在默认环境中安装torch

• 输入 pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
• 输入 pip install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp36-cp36m-win_amd64.whl
  

（4）检测Pytorch-gpu是否安装成功

1. Windows+R，打开cmd

2. 输入 python

3. 逐行输入下面的代码

import torch

torch.__version__

torch.cuda.is_available()

如图所示：

这样，Win10环境下的Pytorch GPU版就安装成功啦！