[ Skip to the content ]

Institute of Formal and Applied Linguistics Wiki

[ Back to the navigation ]

You are here: start » gpu

Differences

This shows you the differences between two versions of the page.

Link to this comparison view

Both sides previous revision Previous revision
Next revision 		Previous revision
Next revision Both sides next revision
gpu [2017/07/28 23:14]
popel [GPU at ÚFAL] 		gpu [2017/08/28 14:40]
kocmanek [Set-up CUDA and CUDNN]
Line 46: Line 46:
  
 In this section will be explained how to use cluster properly.  In this section will be explained how to use cluster properly. 
 +
 +==== Set-up CUDA and CUDNN ====
 +
 +You can add following command into your ~/.bashrc
 +
 +  CUDNN_version=6.0
 +  CUDA_version=8.0
 +  CUDA_DIR_OPT=/opt/cuda-$CUDA_version
 +  if [ -d "$CUDA_DIR_OPT" ] ; then
 +    CUDA_DIR=$CUDA_DIR_OPT
 +    export CUDA_HOME=$CUDA_DIR
 +    export THEANO_FLAGS="cuda.root=$CUDA_HOME,device=gpu,floatX=float32"
 +    export PATH=$PATH:$CUDA_DIR/bin
 +    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_DIR/cudnn/$CUDNN_version/lib64
 +    export CPATH=$CUDA_DIR/cudnn/$CUDNN_version/include:$CPATH
 +  fi
 +
 ==== TensorFlow Environment ==== ==== TensorFlow Environment ====
  

[ Back to the navigation ] [ Back to the content ]