ubuntu 16.04 安装NVIDIA,cuDNN

来源:互联网 发布:天天特价淘宝网毛衣 编辑:程序博客网 时间:2024/06/01 09:12

cuda安装

官网下载cuda:https://developer.nvidia.com/cuda-downloads,我下载的为cuda-8.0版本

cuda安装步骤如下:

  • sudo apt-get install mesa-common-dev freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
  • /etc/modprobe.d/blacklist.conf添加:
  • blacklist amd76x_edac
  • blacklist vga16fb
  • blacklist nouveau
  • blacklist rivafb
  • blacklist nvidiafb
  • blacklist rivatv
  • 更新内核: sudo update-initramfs -u
  • 卸载旧驱动:sudo apt-get remove --purge nvidia*
  • ctrl+alt+f1进入终端界面。 sudo service lightdm stop
  • sudo sh cuda8.0.27linux.run --no-opengl-libs --verbose 进行安装一路enter和y即可。
  • 配置环境变量 在/etc/profile添加:
  • PATH=/usr/local/cuda-8.0/bin:$PATH
  • export PATH
  • 执行 source /etc/profile使其生效
  • 添加库搜索路径。在 /etc/ld.so.conf.d/添加文件cuda.conf,内容为:
  • /usr/local/cuda-8.0/lib64
  • 执行 sudo ldconfig使其生效
  • 执行nvidia-smi测试是否安装成功

cudnn安装:

官网下载cudnn:https://developer.nvidia.com/cudnn

安装cudnn步骤如下:

sudo cp lib* /usr/local/cuda/lib64/

sudo cp cudnn.h /usr/local/cuda/include/

原创粉丝点击