1.安装nviidia-410.78驱动。参考链接。
2.验证开发环境参考链接。
3.下载cuda8。下载链接。
我们按照上述方案进行选择,之后选择下载Base Installer。下载完成之后的话,我们会在ubuntu的下载目录下找到。下载完成之后我习惯将其放入home文件夹中。采用MD5检验序号,看看序号是否会对得上:
禁用nouveau驱动
没有输出表示禁用成功,如果没有禁用成功的话,参考最开始分享的两篇文章。
4.按Alt + ctrl +F1进入命令行界面
关闭图形化界面
sudo service lightdm stop
进入cuda文件终端
sudo chmod +x cuda_10.0.130_410.48_linux.runsudo sh cuda_.xx_xx_linux.run
之后的安装跟之前的这篇文章一样。
4.安装cudnn:/rdp/cudnn-archive。
之后我们将其移入home文件夹下面,然后进入cuda的include文件夹下面,执行下面这个命令复制头文件:
sudo cp cudnn.h /usr/local/cuda/include/
之后在终端进入cuda/lib64文件夹下面,运行以下命令:
#复制动态链接库
sudo cp lib* /usr/local/cuda/lib64/
#删除原有动态文件
cd /usr/local/cuda/lib64/sudo rm -rf libcudnn.so libcudnn.so.6
#生成软衔接
sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6
随后需要将路径/usr/local/cuda/lib64 添加到动态库:
sudo gedit /etc/ld.so.conf.d/cuda.conf
在打开的文件里面输入:
/usr/local/cuda/lib64
终端下接着输入命令使链接生效:
sudo ldconfig
之后查看一下是否链接成功:
sudo ldconfig -v
安装tensorflow1.4.0版本的GPU。
pip install tensorflow-gpu==1.4.0
中间可能会报错:
Cannot uninstall 'six'. It is a distutils installed project and thus we cannot accurately determine which files belong to it which would lead to only a partial uninstall.
解决:
sudo pip install six --upgrade --ignore-installed six
或者你可以参考以下博客:
/@vitali.usau/install-cuda-10-0-cudnn-7-3-and-build-tensorflow-gpu-from-source-on-ubuntu-18-04-3daf720b83fe
我的微信公众号名称:深度学习与先进智能决策
微信公众号ID:MultiAgent1024
公众号介绍:主要研究强化学习、计算机视觉、深度学习、机器学习等相关内容,分享学习过程中的学习笔记和心得!期待您的关注,欢迎一起学习交流进步!