Nvidia cuda cudnn安装

最新推荐文章于 2025-11-23 13:28:02 发布

原创最新推荐文章于 2025-11-23 13:28:02 发布 · 1.5k 阅读

0 ·

CC 4.0 BY-SA版权

ubuntu 同时被 2 个专栏收录

19 篇文章

订阅专栏

Machine Learning

8 篇文章

订阅专栏

本文提供了一种在Ubuntu系统中安装NVIDIA驱动和CUDA的方法，包括删除旧驱动、选择合适的驱动版本、安装CUDA及cuDNN，并对环境变量进行设置。

1. 删除之前驱动: sudo apt-get purge nvidia-*
可能出现问题：
无法获得锁 /var/lib/dpkg/lock - open (11: 资源暂时不可用)
E: 无法锁定管理目录(/var/lib/dpkg/)，是否有其他进程正占用它？”
　解决方法：
　　　　　　　　　　　　　a. 终端：ps -aux # 查看所有进程
　　　　　　　　　　　　　b. 找到apt-get的进程的PID号，直接：sudo kill PID
　　　　　　　　　　　　　c. sudo rm /var/cache/apt/archives/lock
sudo rm /var/lib/dpkg/lock
2. 终端输入：ubuntu-drivers devices　　　　　　# 查看适合电脑的驱动程序
比如driver : nvidia-387 - third-party free recommended　　　　# 找到recommended推荐的版本
sudo apt-get install nvidia-387 # 安装(较慢,可开启清华镜像)
3. 　sudo reboot重启
输入nvidia-settings查看配置情况,nvidia驱动安装成功
4. 官网下载:
cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
cudnn-8.0-linux-x64-v5.0-ga.tgz
5. 下载目录下运行:（确保nvidia驱动安装成功再执行）
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
安装成功会在usr/local下看到cuda-8.0、cuda两个文件夹
6. 解压cudnn-8.0-linux-x64-v5.0-ga.tgz
cd cuda/include/
sudo cp cudnn.h /usr/local/cuda/include/ #复制cudnn.h到/usr/local/cuda/include下,注意这里是复制到cuda下不是cuda-8.0
cd ../lib64
sudo cp lib* /usr/local/cuda/lib64/ # 同理复制lib*的三个文件
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

7. cudnn解压后lib64下有libcudnn.so、libcudnn.so.5.0.5、libcudnn.so.5三个文件夹
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5
sudo ln -s libcudnn.so.5.0.5 libcudnn.so.5
sudo ln -s libcudnn.so.5 libcudnn.so

8. 添加环境变量
sudo gedit /etc/profile
末尾加上:
PATH=/usr/local/cuda/bin:$PATH
export PATH

sudo vim /etc/ld.so.conf.d/cuda.conf
按a进入插入模型输入：
　　　/usr/local/cuda/lib64
然后esc + :wq退出保存即可
sudo ldconfig　　　　　　　# 生效

9. sudo reboot

10. 测试：
　　　　　/usr/local/cuda-8.0/sample下运行:sudo make
　再进入sample/1_Utilities/deviceQuery下：./deviceQuery
　最后输出Result = PASS即可