这篇文章主要是讲安装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
评论(0)
您还未登录,请登录后发表或查看评论