Ubuntu18.04 CUDA与cudnn安装

cuda版本的选择

CUDA与cudnn有多种安装方式,目前搜索cuda下载默认打开的是10.2:https://developer.nvidia.com/cuda-downloads

进入连接,按照要求选择。本人使用的是1070,Ubuntu18.04,驱动为440.59,因此这里直接选择了10.2,如果你不是请参开该网址以下两个网址

  1. https://developer.nvidia.com/cuda-gpus

该网址记载了最低版本要求,比如我的1070,使用于cuda6.1及以上版本。由于Ubunt,18.04 默认gcc版本为7.3,可以直接使用。因此这里不错调整,如需调整请参考——linux下gcc、g++不同版本的安装和切换

gpu-cuda

  1. https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

该网址则展示了cuda于derivers之间的关系。
cuda drivers

cuda的安装

而cuda安装包本身则有多种,因为我们在中国,考虑到服务器的问题。建议选择run file或者deb安装。本人选择的是run file。不过差别不大。

cuda download

在选择后,下方会出现安装时使用的命令。直接复制命令,下载完成后,输入命令运行即可,非常方便。

注意如果已经安装驱动,则不需要再进行安装。其他都可以选择y,安装的cuda编程的例子可以用来做安装验证,方法见最后。如果cuda安装失败,则考虑

ubuntu-drivers devices  #查看自己的显卡及可以安装的驱动版本

如果显示目前使用的显卡为集显,则需要考虑显卡的切换问题,这个请参考该文章——

Ubuntu 18.04 独显和集显切换

无论你选择的什么安装方式,只要不选择显卡驱动安装,然后直接一路安装即可。

CUDA验证

cd /home/fonttian/NVIDIA_CUDA-10.2_Samples/
sudo make
cd /bin/x86_64/linux/release
./deviceQuery

test CUDA

cudnn的安装

cudnn 的下载地址为——https://developer.nvidia.com/cudnn,不过依旧是需要在登录后下载。

cudnn download

登陆后,会出现列表以供选择。如果选择deb包,下载后直接sudo dpkg -i filename.deb即可。

如果下载的是tar文件,则需要解压缩并复制到对应的文件夹。之后修改一下权限即可。

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 /usr/local/cuda/lib64/libcudnn*

cudnn检查

nvidia-smi
# 查看nvidia信息,会显示cuda版本号
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
# 查看cudnn版本

如果是deb安装的cudnn,会在/usr/src/cudnn_samples_v7有一些cudnn的例子。我们可以通过编译mnistCUDNN sample进行验证。

cp -r /usr/src/cudnn_samples_v7/ $HOME
cd  $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN

如果安装成功将会有如下图所示的:Test passed!

(由于本人使用的是tar,因此网上搜了张图片,该图片来自于——ubuntu16.04安装cuDNN的两种方式以及验证
test cudnn

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Font Tian

写的很好,请给我钱

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值