深度学习环境配置:cuda+cudnn→pytorch

简述

cuda是专为GPU设计的软件,其目的是开发GPU的计算能力。cudnn是cuda所需的包。pytorch是为深度学习开发的python库。该库需要cuda的支持。

确定显卡驱动程序

一张显卡需要合适的驱动才能发挥最优性能,配置pytorch的第一件事就是为自己的显卡找到合适的驱动。你的电脑上或许已经有了一个显卡驱动,但我个人认为还是参考一下官方给出的最新版比较保险。点击 GEFORCE® 驱动程序 寻找。
如果官网给出的驱动程序已有更新,则建议下载比较新的。但不一定要最新的,因为相比于最新版本,稍低一点的版本稳定性更高。

安装cuda

找到NVIDIA控制面板,如图1,打开,依次点击帮助、系统信息、组件,找到如图2红圈部分信息,确定支持的cuda版本。
图1-NVIDIA控制面板图2-支持的cuda版本cudaToolkit官网 下载对应版本的cuda,按提示安装。

下载cuddn并配置cuda。

NVIDIA官网 下载cuddn。注意选择与cuda相匹配的版本。下载完成后到路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA 下找到安装好的cuda。将cuddn包中bin、include、lib文件夹下的文件分别拷贝到cuda路径下的同名文件中。注意是分别拷贝。

使用anaconda

anaconda是python包管理工具,利用它可以方便的配置python运行环境。自行下载安装,完成后配置一个python环境。

配置pytorch

命令行输入conda env list 查看python环境
命令行输入 activate envName(环境名) 进入对应环境。
进入pytorch官网,复制cuda对应版本的pytorch安装命令。回到命令行,粘贴,开始运行,等待下载结束。

验证pytorch是否安装成功

命令行依次输入:
activate envName
python
再输入如下代码,查看返回值和注释结果是否相符

import torch
print(torch.cuda.is_available())#结果为true表示安装成功

注:过程中有些细微之处没有详细列出,需要自行补充。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值