当前位置: 代码迷 >> 综合 >> win10-cuda9.1+cuDNN7.1+pytorch1.1-GPU安装
  详细解决方案

win10-cuda9.1+cuDNN7.1+pytorch1.1-GPU安装

热度:97   发布时间:2023-10-15 20:45:36.0

pytorch1.1-GPU安装

对pytorch gpu版本的安装,稍微会复杂些,大概需要以下几个步骤:
(1)安装Nvidia Cuda;
(2)安装CuDNN;
(3)安装GPU版本pytorch;
(4)测试。
博主使用版本资源共享:链接:https://pan.baidu.com/s/18p4IdgfklEvQvo7FV09oVw
(提取码:336k )

具体步骤:
(1)安装Nvidia Cuda
首先检查看自己的电脑是否能安装cuda,右键->NVIDIA控制面板,

win10-cuda9.1+cuDNN7.1+pytorch1.1-GPU安装

图一 点击NVIDIA控制面板

NVIDIA控制面板->帮助->系统信息->组件,如下可看出博主电脑支持的是cuda9.1版本。去官网下载对应版本。

win10-cuda9.1+cuDNN7.1+pytorch1.1-GPU安装
下载完成后,可一直顺点直到安装完成。可在环境变量中加入以下路径,测试cuda是否安装成功,只需要win+R打开cmd,输入nvcc -V,若显示版本号,则成功。

win10-cuda9.1+cuDNN7.1+pytorch1.1-GPU安装
(2)安装CuDNN

进入官网下载CuDNN,需要注册一个账号。可在官网目录中,下载CUDA版本对应的CuDNN,由于cuDNN后来更新后,没有cuda9.1对应的版本,所以博主选择for cuda9.0的版本cuDNN7.1。将CuDNN压缩包解压后,下面的三个文件夹复制到Cuda的对应的安装目录下,注意cudnn其lib文件下x64勿替换cuda其lib下的x64文件,而是将cudnn其lib->x64下的文件复制到cuda其lib下的x64文件里。放进去即可。

win10-cuda9.1+cuDNN7.1+pytorch1.1-GPU安装
(3)安装GPU版本pytorch

根据python版本下载对应的pytorch版本,可以点击这个链接,下载对应版本后,我的是用pip的,我把whl文件放到C:\Windows\System32下,然后pip install torch-1.1.0-cp36-cp36m-win_amd64.whl进行安装。

(4)测试
然后进入python环境下,输入

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

返回结果是True,则PyTorch的GPU安装成功。

  相关解决方案