Ubantu18.04上配置cuda10.0和cudnn

Ubantu系统和显卡驱动是小伙伴帮我安装的,cuda10.0和cudnn是我自己安装的,所以在此只说明cuda10.0和cudnn的安装过程~

 

一、安装cuda

在官网(链接1)下载安装文件:

选择完对应版本之后,点击红框下载,第二个需不需要下载我也不太清楚,我没有下载,暂时没啥影响。然后在下载好的文件路径下输入安装命令:

sudo sh cuda_10.0.130_410.48_linux.run 

接下拉按照步骤选择: accept --> n --> y --> y --> y --> ... ,还有几个enter。

安装完成后,设置环境变量。先在终端输入命令来打开文件:

sudo gedit ~/.bashrc

然后在文件末尾添加:

export CUDA_HOME=/usr/local/cuda-10.0
export LD_LIBRARY_PATH=/usr/lcoal/cuda-10.0/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-10.0/bin:$PATH

保存退出后,在终端运行:

source ~/.bashrc

应该已经安装好啦,检查一下是否安装成功:

nvcc -V

还不太够,再检查一下:

cd /usr/local/cuda-10.0/samples/1_Utilities/deviceQuery

sudo make 

./deviceQuery

据说出现 Result = PASS 就没问题了,如果出现 Result = FAIL 就有问题了,可能是驱动版本和cuda版本不匹配,此处参考链接2

二、安装cudnn

首先,去官网(链接3)下载文件,官网需要注册账号。

解压下载好的文件,打开后可以看到cuda文件夹,

在终端打开解压好的文件夹的目录下,以此输入以下四条指令:

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
 
sudo chmod a+r /usr/local/cuda/include/cudnn.h
 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

最后输入以下指令查看一下cudnn的版本,顺便检查一下有没有安装成功~

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

结果如下,说明安装成功啦~  我的cudnn版本是7.6.4,没毛病~

大功告成~! 记录一下安装过程,供大家参考~ 

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值