DeepMind-Atari-Deep-Q-Learner / debian8 jessie CUDA Install

description

• https://github.com/kuz/DeepMind-Atari-Deep-Q-Learner
• cpuでは学習に100日は要する
• cudaをinstallし高速化する
• 細かい点は省く // 不足があるかもしれません

環境

• debian 8
• cpu : i7 2700k
• mem : 32G
• gpu : GTX750

事前作業

• 参考

  • http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#axzz4XCO7KUtT

• 既存のnvidia driverなどは削除する

Insall driver

install required packages

sudo apt-get install build-essential gcc g++ cmake curl libreadline-dev git git-core libjpeg-dev libpng-dev ncurses-dev imagemagick unzip
# 存在しないpackageは削除して実行

sudo apt-get install linux-headers-$(uname -r)
# 適宜, install されている versionにそってinstall

Install CUDA

DL

• https://developer.nvidia.com/cuda-downloads
  • Linux
  • x86_64
  • ubuntu
  • 16.04
  • runfile

Install

# xを落とす
sudo bash cuda_8.0.44_linux.run
ln -s /usr/bin/g++-4.9 /usr/local/cuda/bin/g++

~/.bashrc

export PATH=/usr/local/cuda/bin:$PATH

Learn

git clone https://github.com/kuz/DeepMind-Atari-Deep-Q-Learner.git
cd DeepMind-Atari-Deep-Q-Learner
sudo bash ./install_dependencies.sh
./run_gpu breakout 

結果

• gpuでは, 約7時間で 1M step
• cpuの10~15倍程度
• 1.5M step頃からまともにプレイし始める

補足

• 学習にGPUを専有させれば, さらに早いのかもしれない
• cpu/gpuともに旧世代なので遅いと思う
• i7 2600kだとcpuが頭打ちのようにも見える. 並列動作していないのもあると思われる.