您的位置:首页 > 编程语言 > Python开发

ubuntu18.04: nvidia驱动 + cuda9.0 + cudnn7.0 + python3.6 + tensorflow1.12 安装教程

2018-12-29 15:30 781 查看

教程根据大神 liningxiao2017: https://www.geek-share.com/detail/2722848219.html 改写, 添加了一些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 https://www.geek-share.com/detail/2722848219.html

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: