教程根据大神 liningxiao2017: https://blog.csdn.net/liningxiao2017/article/details/78653795 改写, 添加了一些ubuntu18的注意事项
一. 官网下载并安装对应的显卡(nvdia)驱动:
官网下载: https://www.nvidia.com/Download/index.aspx?lang=en-us 这里提供.run文件的驱动,且避免循环登陆问题
1.卸载掉原有驱动:sudo apt-get remove --purge nvidia*
2.禁用nouveau:sudo vim /etc/modprobe.d/blacklist.conf
在最后一行添加:blacklist nouveau
Vim操作 插入:’i’ 写入:Esc + ‘:wq!’
或者使用gedit,不过首先需要安装: sudo apt-get install gedit
3.执行:sudo update-initramfs -u
4.重启: sudo reboot
5.重启后执行:lsmod | grep nouveau
没有输出即屏蔽好了
6.安装make:sudo apt-get install make
7.禁用X服务:sudo service gdm3 stop
8.进入命令行界面:Ctrl-Alt+F5
账号密码登陆
9.安装驱动:
sudo chmod a+x NVIDIA-Linux-x86_64-410.78.run
sudo ./NVIDIA-Linux-x86_64-410.78.run –no-opengl-files
后面一定要加上–no-opengl-files
10.完成重启:sudo reboot
二 . 安装cuda9.0的run文件
官网下载:https://developer.nvidia.com/cuda-90-download-archive
1.安装gcc和g++:
cuda8.0和cuda9.0不支持最新版本(7.3)的gcc编译,所以为了保险起见,安装4.8版本的gcc和g++
sudo apt-get install gcc-4.8 g++-4.8
如果已经安装了其他版本的gcc
cd /usr/bin
sudo rm gcc g++
执行:
cd /usr/bin
sudo ln –s gcc-4.8 gcc
sudo ln –s g++-4.8 g++
2.安装cuda9.0:
1.执行:
sudo chmod +x cuda_9.0.176_384.81_linux.run
sudo ./cuda_9.0.176_384.81_linux.run
然后根据命令行提示安装;按 q 退出条款浏览或者按空格直到条款末尾,输入 accept 接受条款;输入 n 不安装nvidia图像驱动,前面已经安装好驱动;输入 y 安装cuda 9.0 toolkit;回车确认 cuda 默认安装路径(/usr/local/ cuda-9.0);输入 n 不安装CUDA 9.0 Samples;结束。
2.在PATH变量中加入/usr/local/cuda-9.0/bin:
Sudo gedit ~/.bashrc
末尾添加
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
3.重启:sudo reboot
4.命令行执行 nvcc -V
查看,不报错即表示安装成功
三、安装cudnn9.0-linux-x64-v7.0.tgz
官网下载:https://developer.nvidia.com/rdp/form/cudnn-download-survey 需要账号登陆
1.复制
cd ~/下载/cuda/include
sudo cp *.h /usr/local/cuda/include/
cd ~/下载/cuda/lib64
sudo cp libcudnn* /usr/local/cuda/lib64/
2.建立软连接
cd ../..
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
sudo ln -s libcudnn.so.7.0.21 libcudnn.so.7
sudo ln -s libcudnn.so.7 libcudnn.so
四、替换python,安装pip
Ubuntu18自带python3.6
sudo rm /usr/bin/python
sudo ln -s /usr/bin/python3.6 /usr/bin/python
安装pip
wget https://bootstrap.pypa.io/get-pip.py --no-check-certificate
sudo python get-pip.py
或者
sudo apt-get install python-pip
五、安装tensorflow1.12
sudo pip install tensorflow-gpu==1.12.0
测试
>>> import tensorflow as tf
>>> tf.__version__
'1.12.0'
>>>
引用: ubuntu16.04:cuda8.0+python3.5+tensorflow1.4+opencv3.2安装说明
链接:liningxiao2017 http://blog.csdn.net/liningxiao2017/article/details/78653795