1. 程式人生 > >ubuntu16.04 + cuda-9.0 + cudnn+ tensorflow+ anaconda+ keras安裝

ubuntu16.04 + cuda-9.0 + cudnn+ tensorflow+ anaconda+ keras安裝

裝好ubuntu後,不需要安裝顯示卡驅動,因為cuda有配套的顯示卡驅動。切記:只限於單顯示卡的使用者
1、安裝cuda

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda-9.0

2、安裝cudnn

tar xzvf cudnn-9.0-linux-x64-v7.1.tgz
cd cuda
sudo cp include/* /usr/local/cuda-9.0/include
sudo cp lib64/* /usr/local/cuda-9.0/lib64

在終端執行gedit .bashrc ,在檔案.bashrc最後一行加上以下:

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

關機重啟
3、安裝anaconda

sh Anaconda3-5.0.0.1-Linux-x86_64.sh

一路回車鍵就行了。
4、安裝tensorflow和keras

pip install tensorflow-gpu==1.8.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install msgpack -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install keras==2.1.6 -i https://pypi.tuna.tsinghua.edu.cn/simple/