参考此文安装成功,想看原版的可以点进去看看。
cuda 9.0 complete installation procedure for ubuntu 18.04 LTS
### 安装步骤 ####
# 验证系统是否支持gpu
# 下载并安装 nvidia cuda toolkit 和 cudnn
# 设置环境变量
# 验证是否安装成功
###
### 验证是否支持cuda
lspci | grep -i nvidia
### 验证是否安装gcc,如安装则显示gcc版本
gcc --version
# 首先获取 PPA repository driver
sudo add-apt-repository ppa:graphics-drivers/ppa
# 安装 nvidia 驱动
sudo apt install nvidia-384 nvidia-384-dev
# 安装其他重要的包
sudo apt-get install g++ freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev
# CUDA 9 需要 gcc 6
sudo apt install gcc-6
sudo apt install g++-6
# 下载cuda安装软件https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda_9.0.176_384.81_linux-run
# 使安装软件可运行,并运行安装软件
chmod +x cuda_9.0.176_384.81_linux.run
sudo ./cuda_9.0.176_384.81_linux.run --override
# 在安装开始的时候参考回答这三个问题,尤其是第二个回答否,不然设置gcc/g++链接时会出错(我的电脑是出错了的)
You are attempting to install on an unsupported configuration. Do you wish to continue?
y(es)
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?
n(o)
Install the CUDA 9.0 Toolkit?
y(es)
# 为gcc/g++设置系统链接
sudo ln -s /usr/bin/gcc-6 /usr/local/cuda/bin/gcc
sudo ln -s /usr/bin/g++-6 /usr/local/cuda/bin/g++
# 设置环境变量
echo 'export PATH=/usr/local/cuda-9.0/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
# 安装cudnn,需要注意的是最新的版本可能不是下面显示的
# 从https://developer.nvidia.com/cudnn下载cuDNN,然后使用如下命令解压文件包
tar -xzvf ${cudnn的路径(可以直接在文件中心找到下载的cudnn,直接右键复制后粘贴过来)}
# 将如下文件从cudnn解压后的文件夹中复制到cuda toolkit 相应路径下
sudo cp -P cuda/include/cudnn.h /usr/local/cuda-9.0/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/
sudo chmod a+r /usr/local/cuda-9.0/lib64/libcudnn*
# 最后验证是否安装成功,验证之前要先重启下电脑
nvidia-smi
nvcc -V