Ubuntu1804安装cuda10和cudnn TensorRT

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

Ubuntu安装Nvidia显卡驱动的操作，我这里就省略了。。。如果不知道如何装英伟达独显驱动，请评论区留言，留言超过十人，我会单独写一篇Ubuntu安装Nvidia显卡驱动的文章。

---

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

先分享一下我的资源

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

cuda的官方下载链接：https://developer.nvidia.com/cuda-toolkit-archive

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

cudnn的安装

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

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

如果为False,那你遇到的问题和我的一样，我参考了这篇文档完美解决：https://blog.csdn.net/u012388993/article/details/102573117这里，先感谢一下那个博主。                                                                       

为什么返回为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、微信的链接：https://github.com/wszqkzqk/deepin-wine-ubuntu