一、到NVIDIA官网下载适合自己的GPU版本驱动
NVIDIA官网下载驱动界面:点SEARCH下载.run文件。
如果有旧的NVIDIA驱动,需要先卸载旧的驱动(现在有驱动先不安了,以后再更新):
二、下载DUDA8.0
进入到下载的文件所在目录
cd ~/Downloads
安装CUDA
sudo sh cuda_8.0.44_linux.run
注意安装过程中提示是否安装驱动的时候选择n,其他都选择是。
添加环境变量到~/.bashrc:
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
更新变量:
source ~/.bashrc
还不确定要不要也更新系统变量并重启:
source /etc/profile
source /etc/environment
验证是否安装成功:
// 如果怕把samples搞坏了那就先搞一个备份,在备份里搞
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
安装cuDNN:
1、下载cuDNN5.1
2、安装cuDNN
进入到文件所在目录
cd ~/Downloads
解压
tar -xvf cudnn-8.0-linux-x64-v5.1.tgz
将解压后的文件拷贝到相应的目录下:
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*