Python offers some modules to communicate with a GPU
Both Tensorflow and Torch have been used on this cluster succesfully.
Some version of python and these modules might not work with the GPU
| Module | Version | Python version |
|---|---|---|
| Tensorflow | 2.17.0 | 3.11.10 |
| Torch | 2.0.1 | 3.9.7 |
With these versions the correct CUDA library is also installed
Please run a simple test with python code to make sure you can communicate with the GPU