Prepare for deep learning based on GPU with Python
OS:
Ubuntu 16.04
Building prerequisities:
make
build-essential
python3-dev
Parallel computing prerequisities:
cuda8.0
cudnn6.0
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:/usr/include:/usr/local/cuda/include
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib:/usr/lib/x86_64-linux-gpu/:usr/local/cuda/lib64
- Python 3.5 packages:
chainer
tensorflow-gpu
cupy
nvidia-cuda-dev
is only required when building cupy
locally.
If python package installation fails, try with sudo pip install --force-reinstall --ignore-installed --no-cache-dir -vvvv
to debug.
Prepare for deep learning based on GPU with Python