ubuntu18.04安装CUDA10.1+cudnn

最新推荐文章于 2025-05-20 21:44:04 发布

小萨1332

最新推荐文章于 2025-05-20 21:44:04 发布

阅读量742

点赞数 1

CC 4.0 BY-SA版权

分类专栏：环境配置

本文链接：https://blog.youkuaiyun.com/wzjwj/article/details/104309743

环境配置专栏收录该内容

2 篇文章

订阅专栏

本文详细介绍了在Ubuntu系统中安装NVIDIA GPU驱动、CUDA及cuDNN的步骤，包括选择合适的驱动版本、配置国内镜像源、执行安装命令、设置环境变量等关键操作，确保深度学习环境的顺利搭建。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

虽然安装了好多次，但有时候仍然需要搜索一下教程，所以自己记录下安装过程，避免踩坑。

1.安装显卡驱动

采用最简单的方法，可以避免其他方法出现在登陆界面循环出不来，直接在software&updates->additional Drivers里面选择驱动版本更新（如果没有，需要将下载下载源换乘国内镜像，推荐清华参考：https://blog.youkuaiyun.com/Justdoforever/article/details/90949841）然后Revert

然后打开终端

nvidia-smi

出现下面界面则成功

2.安装CUDA(CUDA各种版本)

下载后执行命令

sudo sh cuda_10.1.105_418.39_linux.run

后续如果有安装驱动的选项，选择不安装

安装完成后添加环境变量

sudo gedit /etc/bash.bashrc

复制下列内容

export PATH=$PATH:/usr/local/cuda/bin  
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64  
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64

然后重新初始化执行该文件使之生效

source /etc/bash.bashrc
sudo ldconfig

重开终端，执行

nvcc -V

出现cuda版本即安装成功。

3.安装cudnn（下载地址）

需要注册才能下载，选择与CUDA对应的cudnn版本且选择cuDNN Library for Linux（是一个压缩包）

下载好后解压,解压后就是cuda文件夹，将对应文件复制到 cuda目录下

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 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

--over