注:筆者的ubuntu18.04 64bit已經安裝好了顯卡驅動，因此沒有此步操作
1.獲取cuda(https://developer.nvidia.com/cuda-downloads,選擇ubuntu相關的runfile,cuda_9.0.176_384.81_linux.run)
2.安裝cuda
sudo apt-get install gcc-6 g++-6 (cuda 9僅支持gcc-6,而ubuntu18.04 64bit默認版本為gcc-7.3.0)cd /usr/bin sudo mv gcc gcc-7.3.0.back sudo ln -s gcc-6 gcc sudo mv g++ g++-7.3.0.back sudo ln -s g++-6 g++chmod u+x cuda_9.0.176_384.81_linux.run
sudo ./cuda_9.0.176_384.81_linux.run (配置時不要安裝驅動,Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81? (y)es/(n)o/(q)uit: n)
將/usr/local/cuda-9.0/bin加入到環境變量PATH中
將/usr/local/cuda-9.0/lib64添加到/etc/ld.so.conf中,並且運行ldconfig進行配置
3.獲取cudnn(https://developer.nvidia.com/rdp/cudnn-download,需要注冊,選擇archive版本cudnn-9.0-linux-x64-v7.4.1.5.tgz,為什么沒有選擇cudnn for ubuntu18.04這樣的版本?因為這個版本太高,目前tensorflow-gpu只支持到cuda9.0,而cudnn必須與cuda版本匹配)
4.安裝cudnn
tar xvf cudnn-9.0-linux-x64-v7.4.1.5.tgz
cd cuda (沒錯，目錄名為cuda)
sudo cp include/* /usr/local/cuda/include/
sudo cp lib64/* /usr/local/cuda/lib64/
5.安裝python相關軟件
sudo apt-get install python-pip python-dev python3-pip python3-dev
6.安裝tensorflow(tensorflow的版本需要與cuda以及cudnn匹配)
pip3 install tensorflow-gpu==1.12.0
7.測試
#python3
>>> import tensorflow as tf (沒有錯誤信息表明基本安裝ok)
8.導入tensorflow時報錯：cannot import name 'abs'的解決方法
8.1 pip uninstall tensorflow-gpu==1.12.0
8.2 mv ~/.local/lib/python3.6/site-packages/tensorflow ~/.local/lib/python3.6/site-packages/tensorflow.back (注意有可能您的路徑與筆者的不一致)
8.3 pip install tensorflow-gpu==1.12.0