这篇文章主要是讲安装cuda和cudnn的,我会一步一步详细说明。

Ubuntu安装Nvidia显卡驱动的操作,这一步一定要小心小心再小心!因为一不小心,你就需要重装系统了!!我是按照这篇博客进行安装的:Ubuntu18.04安装Nvidia显卡驱动教程_CharlesMoCH的博客-CSDN博客_ubuntu18安装nvidia显卡驱动


我的环境:Ubuntu1804 显卡驱动:435.21 我要安装cuda10 cudnn10.0

先分享一下我的资源

cuda10、cudnn10.0 链接: https://pan.baidu.com/s/1-ww32KqPhuOODiTeb-mgDA 密码: p7ka

cuda的官方下载链接:CUDA Toolkit 11.4 Downloads | NVIDIA Developer

cudnn的官方下载链接:https://developer.nvidia.com/rdp/cudnn-download

安装CUDA

1.安装cuda包:进入到你cuda10的安装包,也就是cuda_10.0.130_410.48_linux.run的所在位置,运行如下命令:

sudo sh cuda_9.0.176_384.81_linux.run

然后一直回车回车,确定,直到出现是否安装Nvidia驱动软件(当然是英文的,什么Nvidia Drivers。。。具体忘了) ,输入N,为啥,这里默认你已经安装过Nvidia驱动了。如果你还没安装Nvidia的驱动,那需要专门单独先安装好了再说。因为要禁用Ubuntu自带的驱动,很麻烦,搞不好需要重装系统。

2. 配置环境变量:

sudo gedit  /etc/profile
#在文件末尾中添加路径,也就是安装目录
export  PATH=/usr/local/cuda-10.0/bin:$PATH
export  LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64$LD_LIBRARY_PATH

这里,执行完sudo gedit /etc/profile命令后,会进入profile编辑页面,在文件最后添加另外两行,保存就可以了。这里可能会提示有WARNING,不用管。

3. 重新启动 :

sudo reboot

在命令行输入nvcc -V来检测一下CUDA是否安装成功了吧!

安装cudnn

1.解压安装包:在cudnn压缩包所在位置打开终端,解压cudnn-10.0-linux-x64-v7.tgz命令如下:

tar -xzvf cudnn-10.0-linux-x64-v7.6.5.32.tgz

执行命令后,你会发现多了一个cuda文件夹,如下图所示,如果 解压过程出错,可以多试几次,如果还不行,那就是压缩包出问题了。

2.复制操作

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的安装命令的截图如下所示:

到这一步,恭喜你安装基本完成了!你可以用TensorFlow的GPU代码来检测一下。

import tensorflow as tf
print(tf.test.is_gpu_available())
#如果返回为true则可以调用GPU

如果为False,那你遇到的问题和我的一样,我参考了这篇文档完美解决:解决Could not load dynamic library ‘libcudart.so.10.0’的问题_arenascat的博客-CSDN博客这里,先感谢一下那个博主。

为什么返回为False呢?主要是因为环境变量没配置好

1、用下边代码新建文件

sudo vi /etc/profile.d/cuda.sh

2、将下边的内容粘贴进去并保存(不会vim的,输入i进入插入模式,粘贴,esc, :wq! )

export PATH=$PATH:/usr/local/cuda/bin
export CUDADIR=/usr/local/cuda

3、再新建一个文件

sudo vi /etc/ld.so.conf.d/cuda.conf

4、同2

/usr/local/cuda/lib64

5、设置链接,最后sudo ldconfig出现某某文件not a symbolic link问题,但不用管,直接重启一下电脑就好了

sudo ldconfig

OK,大功告成了 ,再用代码测试一下吧,肯定返回True!

如果觉得有用的话,给我点个赞鼓励一下吧。

安装TensorRT

下篇文章我会写安装使用TensorRT加速推理的博客,如果感兴趣,请关注我,我们一起成长!

补充一个安装QQ、微信的链接:GitHub - wszqkzqk/deepin-wine-ubuntu: Deepin Wine for Ubuntu/Debian

补充一下ubuntu1804安装cuda10.2的参考博客:

强烈推荐第一个链接,很详细

在Ubuntu系统中搭建gpu版pytorch环境_YChao99的博客-CSDN博客

Linux系统CUDA10.2+CUDNN安装教程_周先森爱吃素的博客-CSDN博客_linux安装cudnn

Ubuntu18.04安装CUDA10.2 + cuDNN v7.6.5_hou9021的博客-CSDN博客_ubuntu18.04安装cuda10.2