Ubuntu16.04下GTX970配置机器学习环境

来源：互联网发布：linux 给用户权限编辑：程序博客网时间：2024/04/29 21:46

首先下载以下三个软件

显卡的驱动：http://www.geforce.cn/drivers

cuda tool kit（需要注册）：https://developer.nvidia.com/cuda-downloads

cuDNN（需要注册）：https://developer.nvidia.com/rdp/cudnn-download

1、安装显卡

ctrl+alt+F1 进入命令行模式，关掉方式：sudo /etc/init.d/lightdm stop，然后开始安装显卡驱动：

直接运雄驱动的*.run文件即可。中间会有提示，选yes没有害处，可能会提示卸载之前的驱动，选yes即可。最后提示安装成功，会回到图像桌面，手动重启一下。

但是好几次都没有成功，最后找到了一个简单的方法，那就是在ubuntu的软件管理里面，找到额外驱动那一项，会看到你的显卡情况，可以直接在这里安装驱动。

2、安装cuda

同样进入命令行，关掉X server服务，运行run文件。在选择是否安装驱动时选择否，其他的选择Yes。

结束之后添加环境变量

export PATH=/usr/local/cuda/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda/lib64/:$LD_LIBRARY_PATH

3、安装cuDNN

这个是把一些文件复制到指定位置即可：

tar xvzf cudnn-7.0-linux-x64-v4-prod.tgzsudo cp cuda/include/cudnn.h /usr/local/cuda/includesudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*4、添加环境变量

否则TF不知道CUDA的东西在哪里，方法：在/etc/environment文件中直接添加下面路径：

:/usr/local/cuda/lib64:/usr/local/cuda

至此差不多就可以跑DNN了!

1 0