对ubuntu搭建深度学习环境的总结

1.安装ubuntu

在第一次安装的需要禁止一个显卡类的,安装界面,按e,将quiet slash —改为:“quiet slash nomodeset”,成功安装后,立刻安装NVIDIA的显卡驱动,需要先将nouveau自带的显卡驱动加入黑名单,具体修改参考这篇文章:

vim /etc/modprobe.d/blacklist.conf
在文件最后加入

blacklist nouveau               
options nouveau modeset=0               

之后安装NVIDIA的显卡驱动就好了,需要注意的是先了解自己硬件所支持的那些显卡驱动,然后再进行下载安装。

2.配置深度学习环境

需要下载anconda,anconda会自己带一个python的版本,以及构建一个虚拟的环境,会接管pip,pip和conda下载的第三方的包都会存在anconda的安装目录下的site-package下,很方便。
下载好后安装cuda,和显卡匹配就行,不是很困难,还有与cuda匹配的cudnn,之后用conda install tensorflow 就可以了,进行简单的测试便可以安装成功。

你可能感兴趣的:(对ubuntu搭建深度学习环境的总结)