ubuntu18.04安装CUDA10.1+cudnn

Ubuntu下GPU驱动与CUDA安装指南

最新推荐文章于 2023-09-14 20:29:13 发布

原创最新推荐文章于 2023-09-14 20:29:13 发布 · 759 阅读

4 ·

CC 4.0 BY-SA版权

环境配置专栏收录该内容

2 篇文章

订阅专栏

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

部署运行你感兴趣的模型镜像

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

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

您可能感兴趣的与本文相关的镜像