zl程序教程

您现在的位置是:首页 >  其他

当前栏目

ubuntu18.04 安装CUDA9.0 + CUDNN7 deb安装

安装 ubuntu18.04 deb
2023-06-13 09:11:40 时间

大家好,又见面了,我是你们的朋友全栈君。

一 安装显卡驱动:

直接在系统软件更新中选择安装:

或者选择PPA源安装,参照:https://blog.csdn.net/new_delete_/article/details/81544438

输入命令测试是否安装成功:

nvidia-smi

二、安装CUDA

1、安装gcc-6 g++-6(由于CUDA 9.0仅支持GCC 6.0及以下版本,而Ubuntu 18.04预装GCC版本为7.3)

sudo apt-get install gcc-6 g++-6

2、修改gcc、g++ 默认版本

cd /usr/bin
sudo rm gcc
sudo ln -s gcc-6 gcc
sudo rm g++
sudo ln -s g++-6 g++

3 去官网下载CUDA9.0 地址:https://developer.nvidia.com/cuda-toolkit-archive

3、安装:

sudo chmod +x cuda_9.0.176_384.81_linux.run 
sudo ./cuda_9.0.176_384.81_linux.run 

如果已经安装驱动,就不要再选择安装驱动了

4、添加环境变量:

打开~/.bashrc

sudo gedit ~/.bashrc

在文件末尾添加以下命令:

export CUDA_HOME=/usr/local/cuda 
export PATH=$PATH:$CUDA_HOME/bin 
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

5、测试是否安装成功:

nvcc -V

三、 安装CUDNN

在官网下载相应的版本(deb包):地址:https://developer.nvidia.com/rdp/cudnn-download(需要注册登陆)

2、安装cudnn的deb包

sudo dpkg -i libcudnn7_7.4.1.5-1+cuda9.0_amd64.deb 

完成

发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/143820.html原文链接:https://javaforall.cn