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