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/09/11 11:26] fucik |
gpu [2017/10/12 13:42] ufal [How to use cluster] |
||
---|---|---|---|
Line 81: | 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 116: | 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: |