KobayashiRui
12/9/2017 - 7:27 PM

Keras

kerasに関することをメモしていく予定

#KerasでDQNを実装する 参考
http://neuro-educator.com/rl2/

Keras インストール

cpu版

pip install --upgrade tensorflow
pip install keras

GPU版

wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb

#cudaバージョンでググッてダウンロードでもオッケー
#ダウンロードする場所はホームディレクトリ
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb
sudo apt-get update
sudo apt-get install cuda

#パスの指定.bashrcに書いたほうがいい!
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}update_8.0.61-1_amd64-deb

#cuda8.0の習性パッチ、これもホームディレクトリ
wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/patches/2/cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64-deb
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64-deb

conda install -c conda-forge keras
conda install -c conda-forge tensoorflow-gpu

参考文献