1. 程式人生 > >ubuntu 18.04安裝tensorflow環境

ubuntu 18.04安裝tensorflow環境

安裝顯示卡驅動

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