...
If GPUs are available and accessible, the output generated will indicate they are being used.
Note that our our system-wide CUDA-enabled TensorFlow and PyTorch versions are only available in the default Python 3 command-line environment (e.g. python3 or python3.8 on the command line). They are not yet available in the global JupyterHub environment that uses the Python 3.9 kernel. If you need a different combination of Python and TensorFlow/PyTorch versions, you'll need to construct an appropriate custom Conda environment (e.g. miniconda3 or anaconda).
Resources
CUDA
Both hfogcomp04 and wilkcomp03 have both CUDA 11.8 and CUDA 12.x installed, under version-specific subdirectories of /usr/local.
...