Ubuntu下安裝tensorflow-gpu（1.7）
阿新 • • 發佈：2018-12-17
一、系統環境
作業系統：Ubuntu（16.04）
顯示卡 ：NVIDIA GeForce GTX 1080ti
二、CUDA安裝
1、下載CUDA Toolkit 9.0
2、安裝CUDA
1）執行：
sudo sh cuda_9.0.176_384.81_linux.run2）按照命令列提示操作（我全部選擇了y），請確定 X Server已經退出，否則會報錯：
X Server 關閉命令：
sudo /etc/init.d/lightdm stop3）安裝log中提示缺少如下包
原因是缺少相關的依賴庫,安裝相應庫就解決了
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev 再次安裝,就不再提示了
sudo sh cuda_9.0.176_384.81_linux.run4）設定環境變數
sudo vi ~/.bashrc在文字末尾增加如下內容
export PATH=/usr/local/cuda-9.0/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH export CUDA_HOME=/usr/local/cuda
環境變數立即生效
sudo source ~/.bashrc
sudo ldconfig 5）檢查cuda是否配置正確
nvcc --version三、CUDNN安裝
1、下載Download cuDNN v7.0.4 (Nov 13, 2017), for CUDA 9.0
2、下載完，解壓cudnn,會出現三個檔案，如下：
tar zxvf cudnn-9.0-linux-x64-v7.tgz
把cudnn兩個資料夾中的檔案，複製到CUDA相應資料夾的下面（cuda的預設安裝路徑：/usr/local/cuda）
cp ./include/* /usr/local/cuda/include/ cp ./lib64/* /usr/local/cuda/lib64/
cudnn安裝ok
四、安裝tensorflow-gpu
1、安裝pip （如果報許可權錯誤，則在命令前加 sudo）
apt-get install -y wget vim
wget -O /tmp/get-pip.py https://bootstrap.pypa.io/get-pip.py
python3 /tmp/get-pip.py
pip install --upgrade pip2、安裝其他libs
RUN pip install -U numpy
RUN pip install -U matplotlib
RUN pip install -U pandas3、安裝tensorflow-gpu （如果報許可權錯誤，則在命令前加 sudo）
pip install -U tensorflow-gpu==1.7 --user參考