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