cuda和cudnn安装的简易教程

第一:根据论文代码pytorch或者tensorflow的版本需求,确定cuda下载版本
pytorch与cuda的版本

第二:cuda下载(使用翻墙软件,更快打开网页)
cuda链接
在这里插入图片描述
个人推荐install Tpye选择deb[local],安装步骤只需要复制Base Install中的命令。(其中,为了更快下载安装文件,可以将第一条和第三条命令中的网址直接复制来下载)

环境变量的配置:
在终端输入:

sudo gedit ~/.bashrc
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

关闭bashrc文件,在终端输入

source ~/.bashrc

第三:cudnn下载,并将cudnn解压。解压后cudnn有lib64和include两个文件夹,将对应文件复制到cuda的相应位置。(一共要复制5个文件)
例如:sudo cp cudnn.h ///include/cudnn.h
sudo cp libcudnn.so ///lib64/libcudnn.so

第四:对复制后的cudnn文件修改其权限。(一共要修改5个文件)
例如在include文件夹内,在终端输入:
sudo chmod 777 cudnn.h
注:如果需要安装cudnn8.x,可参考该教程

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值