(1)首先要安装NVIDIA显卡驱动,特别对于新电脑,再找不到nvidia-smi指令时:
打开命令窗口(ctrl+Alt+T)
sudo apt-get purge nvidia*
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-384 nvidia-settings (sudo apt-get install nvidia-410 nvidia-settings)
安装完毕后,运行nvidia-smi,查看生效的显卡驱动。若不显示,重启一下再输。
(2)安装CUDA
下载CUDA9.0版本(runfile),命令行输入(版本各异,使用tab去补全):
sudo bash cuda_9.0.176_384.81_linux.run
(3)安装cuDNN
下载cuDNN for CUDA9.0(压缩包),解压后会得到cuda文件夹(注意目录不能带中文,否则解压失败)。命令行输入:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
(4)修改path
命令行输入:
sudo gedit ~/.bashrc
打开文件后,在最后面添加:
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
(5)命令行输入:
source ~/.bashrc
(6)输入nvcc -V,不出其他毛病的话,可以看到cuda版本。