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 

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章