Ubuntu 18.04安装CUDA和cuDNN,以及驱动卸载重装

编程入门 行业动态 更新时间:2024-10-28 00:18:51

Ubuntu 18.04安装CUDA和cuDNN,以及驱动卸载<a href=https://www.elefans.com/category/jswz/34/1762287.html style=重装"/>

Ubuntu 18.04安装CUDA和cuDNN,以及驱动卸载重装

Ubuntu 18.04安装CUDA和cuDNN,驱动卸载重装

 

操作系统环境:Ubuntu 18.04 LTS


1、安装N卡驱动

首先我们需要添加源,

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

然后检查可以安装的驱动版本,

ubuntu-drivers devices

选择最合适的版本安装即可,

sudo apt install nvidia-driver-XXX

最后,重启Ubuntu系统,输入nvidia-smi查看驱动信息。


2、安装CUDA

进入nvidia开发者网站的CUDA下载页面:

CUDA Toolkit Archive​

developer.nvidia

 

选择runfile格式的CUDA文件下载,

下载完成后,解压,并运行上图中的命令,会有条款,接受即可,注意安装CUDA的时候不要安装驱动(因为在第一步我们已经安装过了)。

然后,

sudo vim ~/.bashrc

我们在文件最后一行添加:

$ export PATH=/usr/local/cuda-10.1/bin:/usr/local/cuda-10.1/NsightCompute-2019.1${PATH:+:${PATH}}
$ export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

最后,使刚才的配置生效。

source ~/.bashrc

进入/usr/local/cuda-10.1/samples/1_Utilities/目录,

sudo make
cd deviceQuery
./deviceQuery

出现如下输出,则CUDA安装成功。


3、安装cuDNN

进入下载页面

NVIDIA cuDNN​developer.nvidia

选择合适的版本下载(cuDNN Library for Linux),

然后解压,并进入到相应目录,运行以下命令:

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*

以配置cuDNN环境。

通过

cat /usr/local/cuda-10.1/include/cudnn.h | grep CUDNN_MAJOR -A 2

可以查看cuDNN版本。

 

4,ubuntu 卸载nvidia驱动 ,安装驱动

 

 

 

 

sudo apt-get --purge remove nvidia*

sudo apt autoremove

sudo apt-get remove --purge nvidia-\*

安装

更多推荐

Ubuntu 18.04安装CUDA和cuDNN,以及驱动卸载重装

本文发布于:2024-03-07 14:57:54,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1718135.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:重装   Ubuntu   cuDNN   CUDA

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!