Skip to content

Instantly share code, notes, and snippets.

@imneonizer

imneonizer/docker-compose-test-gpu.yml

Last active May 7, 2023 15:02
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save imneonizer/e0013a8b8b0a124ea2c355dd66d9ceec to your computer and use it in GitHub Desktop.
Save imneonizer/e0013a8b8b0a124ea2c355dd66d9ceec to your computer and use it in GitHub Desktop.
Download ZIP
Nvidia GPU test using docker-compose
Raw
docker-compose-test-gpu.yml
version: "3.8"
services:
app:
image: nvidia/cuda:11.6.2-base-ubuntu20.04
command: nvidia-smi
deploy:
resources:
reservations:
devices:
- capabilities: [ gpu ]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment