A basic training run using HuggingFace to stress a GPU and test it.

This needs Python along with a few HuggingFace libraries as below

pip install transformers[torch] datasets

The Docker image is uploaded to DockerHub as jakelever/gputest. To run it with GPUs, use the command below

docker run --gpus all jakelever/gputest

The Docker image was created using the Dockerfile and these commands to push it to DockerHub

docker build -t jakelever/gputest .
docker push jakelever/gputest