CUDA+CUDNN准备:
1.安装NVIDIA驱动
(1)查询NVIDIA驱动
首先去官网(http://www.nvidia.com/Download/index.aspx?lang=en-us)查看适合自己显卡的驱动(下载runfile文件):

2)安装驱动
先按Ctrl + Alt + F1到控制台,关闭当前图形环境
$sudo service lightdm stop
卸载可能存在的旧版本 nvidia 驱动
sudo apt-get remove --purge nvidia*
安装驱动可能需要的依赖 $sudo apt-get update $sudo apt-get install dkms build-essential linux-headers-generic sudo nano /etc/modprobe.d/blacklist-nouveau.conf 在文件 blacklist-nouveau.conf 中加入如下内容: blacklist nouveau options nouveau modeset=0 保存退出,执行 sudo update-initramfs -u
ctrl+O保存。然后回车 ctrl+X 退出
reboot重启
运行驱动安装文件
重启后再次进入字符终端界面(Ctrl + Al

本文详细介绍了在Ubuntu系统上安装NVIDIA驱动、CUDA 8.0和cuDNN的过程,包括驱动安装、配置环境变量、安装TensorFlow等深度学习框架,并提供了Python环境设置和测试方法。
最低0.47元/天 解锁文章
725





