diff --git a/services/dask-sidecar/src/simcore_service_dask_sidecar/utils.py b/services/dask-sidecar/src/simcore_service_dask_sidecar/utils.py index 910bd7f4632..882b8d65d71 100644 --- a/services/dask-sidecar/src/simcore_service_dask_sidecar/utils.py +++ b/services/dask-sidecar/src/simcore_service_dask_sidecar/utils.py @@ -1,7 +1,8 @@ import asyncio import logging import uuid -from typing import Any, Awaitable, Coroutine, cast +from collections.abc import Awaitable, Coroutine +from typing import Any, cast import aiodocker from aiodocker.containers import DockerContainer @@ -16,8 +17,8 @@ def _wrap_async_call(fct: Awaitable[Any]) -> Any: def _nvidia_smi_docker_config(cmd: list[str]) -> dict[str, Any]: return { - "Cmd": ["nvidia-smi"] + cmd, - "Image": "nvidia/cuda:10.0-base", + "Cmd": ["nvidia-smi", *cmd], + "Image": "nvidia/cuda:12.2.0-base-ubuntu22.04", "AttachStdin": False, "AttachStdout": False, "AttachStderr": False,