Small script to test that PyTorch works with the current GPU(s).
- Clone the repo:
git clone https://github.com/GRAAL-Research/pytorch-gpu-testing.git. - Launch the script:
./start.sh. The output should indicate the presence of at least one GPU and a training should work with the GPU.