For an interactive session:
docker run -it --rm --network=host --runtime=nvidia --gpus all --shm-size=8g -v <you-directory-here>:/app smellslikeml/gpt-j-training:latest /bin/bash
Content type
Image
Digest
sha256:e8a7800ed…
Size
5.7 GB
Last updated
over 3 years ago
docker pull smellslikeml/gpt-j-training