ubuntu16.04 安裝tensorflow(GPU、CPU)
連結:
https://mirrors.tuna.tsinghua.edu.cn/help/tensorflow/
【注:需要更改成清華源】
GPU安裝:
適用GTX1050,GTX1060,GTX1070,GTX1080等高階顯示卡系列
- 安裝好系統後,更新驅動。系統設定-》軟體和更新-》附加驅動-》更新nvidia和intel
- 安裝cuda9
下載 cuda9.run,(deb格式容易出錯誤)。執行如下命令:
sudo sh cuda_9.0.176_384.81_linux.run
accept
n(不要安裝driver)
y
y
y
安裝完成後,設定環境變數。
設定環境變數:
ctrl+H 開啟 (主目錄/home/*/) .bashrc檔案,新增
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH
##############注意 解決import tensorflow 錯誤的問題##############
更改為:(這裡執行,不會再出現錯誤。很抱歉。)
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
終端執行:source ~/.bashrc
檢查:nvcc --version
3. 安裝cudnn7
下載 for linux版本的,執行如下命令:
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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
4. 安裝anaconda(python3.5)
bash Anaconda3-4.2.0-Linux-x86_64.sh
yes(設定環境變數)
5. 安裝tensorflow-gpu(任意版本)
複製開始的藍色連結上的命令。
CPU安裝:
直接複製開始連結上的命令安裝tensorflow-cpu。