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

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控制面板,

点击NVIDIA控制面板

图一 点击NVIDIA控制面板

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

在这里插入图片描述
下载完成后,可一直顺点直到安装完成。可在环境变量中加入以下路径,测试cuda是否安装成功,只需要win+R打开cmd,输入nvcc -V,若显示版本号,则成功。

在这里插入图片描述
(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文件里。放进去即可。

在这里插入图片描述
(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安装成功。

  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值