1. 程式人生 > >ubuntu16.04 + Anaconda3 + tensorflow-gpu1.8的安裝總結

ubuntu16.04 + Anaconda3 + tensorflow-gpu1.8的安裝總結

需要安裝

CUDA,cuDNN

那麼除了換源,還可以:

設定超時時間

pip --default-timeout=100 install -U Pillow

記得環境下,目前python3.7不能安裝tensorflow,最好降到3.6

官網要求1.8需要 cuda 9.0(9.1都不行) cuDNN7以上

但其實安裝過程中不需要cuDNN版本要求 ,我的3.0.02都成功了

出現:

錯誤:ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

不過需要

vim ~/.bashrc(或者~/.zshrc)最後加入下面環境變數

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64 export PATH=$PATH:/usr/local/cuda-9.0/bin export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0  

別忘了source  ~/.bashrc一下哦~