pytorch-gpu-testing

Small script to test that PyTorch works with the current GPU(s).

Instructions

1. Clone the repo: git clone https://github.com/GRAAL-Research/pytorch-gpu-testing.git.
2. Launch the script: ./start.sh. The output should indicate the presence of at least one GPU and a training should work with the GPU.