ubantu安装cuda和cudnn
作者:互联网
文章目录
首先,保证有一个显卡,并且工作正常,狗头。
1. 安装nvidia驱动;
执行下面命令,系统会告诉你,需要什么驱动;
ubuntu-drivers devices
执行下面命令,安装系统推荐驱动;
sudo ubuntu-drivers autoinstall
安装完成后,执行命令:
nvidia-smi
查看显卡的工作情况;
2. 安装cuda;
选择适合的cuda版本,这点往往是由你要执行的任务决定的。
下载连接:
https://developer.nvidia.com/cuda-downloads
选择你的型号,按照说明执行安装即可,建议选择runfile。
如图,执行对应命令,安装完成。
这里,安装cuda10.02,ubantu版本,runfile(local)。
执行如下命令:
wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run
如果报gcc版本冲突,可能需要gcc降级,参考连接:ubantu 20.04 gcc和cuda版本冲突
具体细节,可参考Ubuntu 18.04安装CUDA和cuDNN。
3.安装cudnn
下载链接,cudnn下载链接,这个需要登录验证,选择对应的版本即可,。可参考链接:ubantu安装CUDA和cudnn,但这个可能在查看cudnn版本时有一个坑。
这里贴出安装的命令:
在下载并解压后,在解压后的文件夹内终端执行:
sudo cp cuda/include/* /usr/local/cuda-10.2/include/
sudo cp cuda/lib64/* /usr/local/cuda-10.2/lib64/
sudo chmod +x /usr/local/cuda-10.2/include/cudnn.h
sudo chmod +x /usr/local/cuda-10.2/lib64/libcudnn*
完成后,查看版本:
可以尝试
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
没有显示,可以执行如下命令:
cat /usr/local/cuda/include/cudnn_version.h|grep CUDNN_MAJOR -A 2
操作截图:
安装完成。
标签:ubantu,local,cudnn,10.2,cuda,安装,usr 来源: https://blog.csdn.net/RookieWantToFly/article/details/120925650