compute_ru

This is an old revision of the document!

Using GPU with python

Python offers some modules to communicate with a GPU
Both Tensorflow and Torch have been used on this cluster succesfully.

An example setup using Python venv, run these commands
python3 -m venv myTorch
source myTorch/bin/activate
pip install torch
The new Python virtual environment has been created and activated, torch package installed
And the correct CUDA library is also installed
Please run a simple test with python code to make sure you can communicate with the GPU

Those version have also been used before and known to work

Module	Version	Python version
Tensorflow	2.17.0	3.11.10
Torch	2.0.1	3.9.7