Ubuntu18.04系统环境下TensorFlow-GPU环境搭建流程

        因为之前一直用的Windows对Linux不是很熟所以在第一次操作的时候踩了不少坑,最后重新刷了一边系统重头开始才一切OK。接下来就把Ubuntu从初始的系统到安装好深度学习环境的流程及其中的坑点进行一下梳理(以备查阅)。

        闲话不多说,以下全部流程建立在已经安装好Ubuntu的基础上。

 

(1)安装Nvidia显卡驱动(坑点,加粗

首先Ubuntu18.04在刚安装好系统时有个问题就是包括点击设置等在内诸多操作都会卡死,查了一下是因为没有显卡驱动,第一次的时候本人就盲目的按网上查到的去自动安装了一个390版本的驱动(坑点之一)。而众所周知截至目前20190522为止,TensorFlow新版需要的显卡驱动是要410以上的。不光如此,在自动安装了显卡驱动后可以再去安装驱动(接下来会说),也可以安装成功,但是在安装CUDA时会报错,无法安装(就是这个点导致本人最终重刷了系统,所幸需要的步骤不多,单纯为了深度学习且网络不差的话小半天足够)。

第二次有了前车之鉴,这次上来没有着急先去安装一个显卡驱动而是直接一步就位,具体参考的是Ubuntu 18.04 NVIDIA驱动安装总结 这篇文章。以下记录下本人采用的PPA方式(感觉这种最简单)

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
ubuntu-drivers devices
sudo apt install nvidia-driver-430

就四条命令,第一条是添加驱动的仓库,第二条更新一下列表,第三条列出适合本机的驱动列表,第四条是安装,其中最后的 nvidia-driver-430根据列表显示的名称自行选择(会有一个recommond的,本机430就是推荐的驱动版本)。以上都搞定等就完了(下载安装驱动),最后好了可以重启一下电脑,重启之后就可以去点设置之类的了也不会卡死。

(2)安装CUDA(也有坑

其实以下已经可以参考TensorFlow上的去进行了,而且GPU支持也列出了所需的软件版本信息。但是这里还是建议去Nvidia官网上用官网的流程进行安装(因为粘贴使用TensorFlow上的那堆指令最后部分出了错没起作用)。CUDA下载链接CUDA的话已经有了10.1,但是按照TensorFlow的支持情况还是要去下载安装10.0版本的,所以进去后要选择Legacy Releases(过往版本),选择好对应平台后安装方式建议选network的在线安装,反正都是下载为用离线版的在指令的第二步倒入pub的时候各种提示找不到文件,当然这可能是我个人原因,所以按心情选就好,反正差不多的流程。

官网上在下载页面的下面直接就有对应的指令

sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb

sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

sudo apt-get update

sudo apt-get install cuda-10.0

前三步都和官网一样正常安就可以了,第四步注意,这里藏了坑,如果直接按官网指令去安cuda的话因为10.1的存在默认是去安10.1的(那我之前选什么Legacy Releases阿!),所以需要在cuda后面加个-10.0告诉它我们就要10.0就好,谢谢。这里在线安装需要下载1.5G左右的文件还是等,如果按照之前的流程先安装好了高版本的驱动这里是不会i有问题的。安装好CUDA后直接把对应版本的cuDNN解压到对应的文件夹即可。

sudo tar -zxvf cudnn-10.0-linux-x64-v7.5.1.10.tgz -C /usr/local

可以直接用这个指令解压过去,没特殊情况都是这个位置。 在cuDNN也解压好之后在PATH里加一下路径

vim ~/.bashrc打开文件,按i进入编辑直接在最下方粘贴上下面的代码即可:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64
export CUDA_HOME=/usr/local/cuda
export PATH=/usr/local/cuda/bin:$PATH

粘贴好后大写ZZ离开,推出后source ~/.bashrc更新文件。以上就基本完成了坑点添坑。

(3)anaconda3安装需要的Python库

这里就记录一条:终端输入anaconda-navigator可打开图形界面(懒人离不开GUI)

(4)代码工具个人推荐VScode

Windows上用的还挺习惯的,最关键一点是:安装使用超简单!直接去官网下个.deb回来双击运行安装,安装好后会出现在全部应用里,去里面给它添加到收藏夹以后用起来就更方便了

你可能感兴趣的:(学习进程)