ubuntu下安装CUDA

/

软硬件环境

  • ubuntu 16.04 64bit
  • NVidia GTX 1070
  • CUDA 9.0

安装驱动

从官网下载RUN文件,首先屏蔽开源nouvear驱动,编辑/etc/modprobe.d/blacklist-nouveau.conf文件,加入内容

  1. blacklist vga16fb
  2. blacklist nouveau
  3. blacklist rivafb
  4. blacklist nvidiafb
  5. blacklist rivatv

如果你原来安装过驱动,最好先删除

  1. sudo apt-get --purge remove nvidia-*
  2. sudo apt-get --purge remove xserver-xorg-video-nouveau

NVIDIA的驱动无法在GUI的环境下安装,所以得切换到命令行,按组合快捷键Ctrl + Alt + F1

  1. sudo /etc/init.d/lightdm stop
  2. sudo ./NVIDIA-Linux-x86_64-390.25.RUN

安装过程基本上就是选中yes or no,装完之后执行

  1. sudo update-initramfs -u

安装CUDA

同样的,我们也进入命令行模式安装cuda

  1. sudo /etc/init.d/lightdm stop
  2. sudo ./cuda_9.0.176_384.81_linux.run
  3. sudo /etc/init.d/lightdm start

然后配置下,编辑~/.bashrc

最后

  1. source ~/.bashrc

安装cudnn

到官网https://developer.nvidia.com/cudnn去下载,这里需要注册,有点麻烦

  1. tar xvf cudnn-9.1-linux-x64-v7.1.tgz
  2. cd cuda
  3. sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
  4. sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64
  5. sudo chmod a+r /usr/local/cuda-9.0/lib64/libcudnn*

测试

安装完毕,我们来测试下环境是否都ok

刚好我这里跑了示例,所以你能看到gpu基本上是满负荷工作

参考资料

http://www.nvidia.com/Download/index.aspx?lang=en-us
https://developer.nvidia.com/cuda-toolkit
https://blog.csdn.net/hanging_gardens/article/details/64439344

转载请注明作者和出处,并添加本页链接。
原文链接: blog.xugaoxiang.com/117

给我留言