diff --git a/compose.yml b/compose.yml index f40aae0..273b0ef 100644 --- a/compose.yml +++ b/compose.yml @@ -66,7 +66,7 @@ services: - ENABLE_SERVICE_GLANCES=1 # enable resources monitor - ENABLE_SERVICE_TENSORBOARD=1 # enable tensorflow training tracker - TF_CPP_MIN_LOG_LEVEL=3 # make tensorflow less verbose - - NVIDIA_AUTODETECT_IMAGE=nvidia/cuda:12.9.1-base-ubuntu24.04 # image with `nvidia-smi` for gpu autodetection + - NVIDIA_AUTODETECT_IMAGE=nvidia/cuda:13.0.0-base-ubuntu24.04 # image with `nvidia-smi` for gpu autodetection labels: # Enable proxy support from Traefik - "traefik.enable=true"