Win10+CUDA9.0+Pytorch+Py3.6-GPU环境搭建

• 第一步、首先看下自己的显卡型号支持什么版本的CUDA利用NVIDA控制面板 见下图

• 

• 第二步：到官网下载对应CUDA和CUDNN

• https://developer.nvidia.com/cuda-toolkit-archive

• 

• https://developer.nvidia.com/rdp/cudnn-archive

• 

• CUDA环境变量设置

• 

• 以上安装后自动生成

• path下添加以下路径

• 

• 这里的CUDNN具体移动复制步骤 

• CUDNN解压，把解压出来的中bin,include,lib中的文件分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0路径下！

• 把C:\Program Files\NVIDIA GPUComputing Toolkit\CUDA\v9.0\extras\CUPTI\libx64里面的cupti64_80.dll复制到C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin路径下

• 第三步：安装对应版本的torch  如果版本不统一 CUDA 调用将返回FALSE 本文安装的是对应于CUDA9.0的torch 执行以下操作即可

  pip install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl

  pip install torchvision

• 第四步：测试

  import torch
  print(torch.cuda.is_available())

• return True 即完成GPU版本安装

• 若以上步骤都没问题则可能电脑设置使用的集成显卡可参照下图查看修改 选择独显即可

•