小規模計算サーバ - arosh/arosh.github.com GitHub Wiki

How to set environment variables

module load python-3.5 && module load cuda-7.0 && source tesla/kerasenv/bin/activate

THEANO_FLAGS

http://deeplearning.net/software/theano/library/config.html

THEANO_FLAGS='device=gpu,lib.cnmem=1' python cifar10_cnn.py

Check GPU Usage

nvidia-smi

nvidia-smi -l updates screen in each 5 seconds.

How to use environment modules

Examples of modulefiles

.modulefiles/python-3.5

#%Module 1.0
set python_home $::env(HOME)/tesla/python3.5
prepend-path PATH $python_home/bin
prepend-path LD_LIBRARY_PATH $python_home/lib

.modulefiles/cuda-7.0

#%Module 1.0
set cuda_home /opt/nvidia/cuda-7.0
prepend-path PATH $cuda_home/bin
prepend-path LD_LIBRARY_PATH $cuda_home/lib64