...
To verify GPU access, use either the rocm-smi (AMD GPU servers) or nvidia-smi (NVIDIA GPU servers).
Two Python scripts are located in /stor/scratch/GPU_info that can be used to ensure you have access to the server's GPUs from TensorFlow or PyTorch. You can run them from the command line like this:
- Tensor Flow
- python3 /stor/scratch/GPU_info/tensorflow_example.py
- PyTorch
- python3 /stor/scratch/GPU_info/pytorch_example.py (test script not yet working on AMD servers)
- python3 /stor/scratch/GPU_info/pytorch_example.py (test script not yet working on AMD servers)
If GPUs are available and accessible, the output generated will indicate they are being used.
GPU-enabled software
See server-specific pages for how to run TensorFlow, PyTorch and AlphaFold.
...