Differences
This shows you the differences between two versions of the page.
Both sides previous revision Previous revision Next revision | Previous revision Next revision Both sides next revision | ||
gpu [2017/08/28 14:46] kocmanek [Set-up CUDA and CUDNN] |
gpu [2017/10/12 13:42] ufal [How to use cluster] |
||
---|---|---|---|
Line 11: | Line 11: | ||
| twister1; twister2; kronos | Tesla K40c | cc3.5 | | | twister1; twister2; kronos | Tesla K40c | cc3.5 | | ||
| dll1; dll2 | GeForce GTX 1080 | cc6.1 | | | dll1; dll2 | GeForce GTX 1080 | cc6.1 | | ||
- | | titan | GeForce GTX 1080 Ti | cc6.1 | | + | | titan | GeForce GTX 1080 | cc6.1 | |
| dll3; dll4; dll5 | GeForce GTX 1080 Ti | cc6.1 | 10| 11 GB | dll3 has only 9 GPUs since 2017/07 | | | dll3; dll4; dll5 | GeForce GTX 1080 Ti | cc6.1 | 10| 11 GB | dll3 has only 9 GPUs since 2017/07 | | ||
+ | | dll6 | GeForce GTX 1080 Ti | cc6.1 | | ||
Desktop machines: | Desktop machines: | ||
Line 59: | Line 60: | ||
export THEANO_FLAGS=" | export THEANO_FLAGS=" | ||
export PATH=$PATH: | export PATH=$PATH: | ||
- | export LD_LIBRARY_PATH=$LD_LIBRARY_PATH: | + | export LD_LIBRARY_PATH=$LD_LIBRARY_PATH: |
export CPATH=$CUDA_DIR/ | export CPATH=$CUDA_DIR/ | ||
fi | fi | ||
Line 80: | Line 81: | ||
This environment have TensorFlow 1.0 and all necessary requirements for NeuralMonkey. | This environment have TensorFlow 1.0 and all necessary requirements for NeuralMonkey. | ||
+ | |||
+ | ==== Pytorch Environment ==== | ||
+ | |||
+ | If you want to use pytorch, there is a ready-made environment in | ||
+ | |||
+ | / | ||
+ | | ||
+ | It does rely on the CUDA and CuDNN setup above. | ||
==== Using cluster ==== | ==== Using cluster ==== | ||
Line 115: | Line 124: | ||
/ | / | ||
# shows CUDA capability etc. | # shows CUDA capability etc. | ||
+ | ssh dll1; ~popel/ | ||
+ | # who occupies which card on a given machine | ||
| | ||
=== Select GPU device === | === Select GPU device === | ||
- | Use variable CUDA_VISIBLE_DEVICES | + | The variable CUDA_VISIBLE_DEVICES |
- | export CUDA_VISIBLE_DEVICES=0 | + | |
To list available devices, use: | To list available devices, use: |