查看Ubuntu14.04 64位上显卡信息,执行:
lspci | grep -i vga
lspci -v -s 01:00.0
nvidia-smi
第一条此命令可以显示一些显卡的相关信息;如果想查看某个详细信息,可以执行第二条命令;如果是NVIDIA卡,可继续执行第三条命令,显示为GeForce GTX 970。
参考https://wiki.ubuntu.com.cn/NVIDIA安装GeForce GTX 970驱动。
安装cuda 8.0:
1.参考官网: http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#axzz4TcLN9lqd
2.从https://developer.nvidia.com/cuda-downloads下载cuda8,x86_64,Ubuntu,14.04
3.依次执行:前面的命令是为了验证主机是否满足cuda8.0安装的要求
lspci | grep -i nvidia
uname -m && /etc/*release
gcc --version
uname -r
sudoapt-get install linux-headers-$(uname -r)
sudo sh cuda_8.0.27_linux.run
执行cuda_8.0.27_linux.run结果如下图所示:第一个输入accept,第二个输入n,其余输入y或者直接回车
4.设置环境变量,执行:
sudo vim /etc/profile
在文件末尾添加,如下内容:
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64/:/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-8.0/bin:/usr/local/cuda/bin:$PATH
5.验证cuda8是否安装成功:
(1)、进入NVIDIA_CUDA-8.0_Samples目录,执行:$ make
(2)、进入1_Utilities目录,执行:$ ./deviceQuery/deviceQuery ,结果如下:
(3)、执行:$ nvcc -V,结果如下:
6.安装cudnn5.0:从https://developer.nvidia.com/rdp/cudnn-download 下载cudnn-8.0-linux-x64-v5.0-ga.tgz
(1)、解压缩:会生成cuda/include和cuda/lib64
(2)、将cuda/include目录中的cudnn.h文件拷贝到/usr/local/cuda-8.0/include/目录下
(3)、将cuda/lib64目录中的库拷贝到/usr/local/cuda-8.0/lib64/目录下;
(4)、将/usr/local/cuda-8.0/lib64/目录下的libcudnn库的权限改为777,执行:
sudo chmod 777 libcudnn*
到此cuda8.0+cudnn5.0就成功安装到ubuntu 14.04 64位机上了,安装步骤比较麻烦些。