ubuntu 18.04安装tensorflow环境教程
参考:https://blog.csdn.net/tjuyanming/article/details/80862290
https://blog.csdn.net/u010801439/article/details/80483036
安装显卡驱动
1、禁nouveau
$ sudo gedit /boot/grub/grub.cfg
在文本中搜索quiet slash
然后添加acpi_osi=linux nomodeset
,保存文本即可。
2、使用PPA仓库进行自动安装
$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt update
\#识别显卡模型和推荐的驱动程序
$ ubuntu-drivers devices
$ sudo apt install nvidia-390
一旦完成,即可重新启动系统。
安装cudn
GCC降级
sudo apt-get install gcc-4.8
sudo apt-get install g++-4.8
装完后进入到/usr/bin目录下
$ls -l gcc*
发现gcc链接到gcc-7.0, 需要将它改为链接到gcc-4.8,方法如下
sudo mv gcc gcc.bak #备份
sudo ln -s gcc-4.8 gcc #重新链接
ls -l g++*
sudo mv g++ g++.bak
sudo ln -s g++-4.8 g++
再查看gcc和g++版本号:
gcc -v g++ -v
均显示gcc version 4.8 ,说明gcc 4.8安装成功。
安装cuda ,及其补丁
sudo sh cuda\_9.1.85\_387.26\_linux.run
sudo sh cuda\_9.1.85.1\_linux.run
sudo sh cuda\_9.1.85.2\_linux.run
sudo sh cuda\_9.1.85.3\_linux.run
安装完毕之后,将以下两条加入.bashrc
文件中.
sudo vim ~/.bashrc
export PATH=/usr/local/cuda-9.1/bin${PATH:+:$PATH}}
export LD\_LIBRARY\_PATH=/usr/local/cuda-9.1/lib64${LD\_LIBRARY\_PATH:+:${LD\_LIBRARY\_PATH}}
安装cuDNN(针对18.04)
tar -xzvf cudnn-9.0-linux-x64-v7.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
sudo apt-get install python-pip python-dev
sudo pip install --upgrade pi
pip install tensorflow-gpu