1、装显卡驱动
检测显卡型号和对应推荐安装驱动
root$ ubuntu-drivers devices
自动安装所推荐的驱动
root$ sudo ubuntu-drivers autoinstall
确保BIOS中Secure Boot设置为disabled,
重启检查驱动是否安装成功
root$ nvidia-smi
2、装CUDA
链接:https://developer.nvidia.com/cuda-toolkit-archive
CUDA推荐下载.run可以根据提示安装
执行如下命令:
sudo sh cuda_9.0.176_384.81_linux.run
然后用一本书的书角压住回车键,直到服务条款显示到100%。接着按下面的步骤选择:
accept
n(不要安装driver)
y
y
y
安装完成后,设置环境变量。
(3)设置环境变量。
打开主目录下的 .bashrc文件添加如下路径,例如我的.bashrc文件在/home/wangyuanwei下,如果没有找到,则按Ctrl+H键显示隐藏文件。
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
终端运行:source ~/.bashrc
检查:nvcc --version
3.cudnn的安装
cudnn的安装非常简单,
(1)下载安装文件
按需求下载cudnn的安装文件:https://developer.nvidia.com/rdp/cudnn-archive
(2)安装cudnn
解压下载的文件,可以看到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*
(3)查看cudnn版本
在终端输入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2