Skip to content

Commit 4ecc58b

Browse files
authored
šŸ›upgrade dask-sidecar cuda version to properly detect GPU nodes (#4466)
1 parent d7f1575 commit 4ecc58b

File tree

1 file changed

+4
-3
lines changed
  • services/dask-sidecar/src/simcore_service_dask_sidecar

1 file changed

+4
-3
lines changed

ā€Žservices/dask-sidecar/src/simcore_service_dask_sidecar/utils.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
import asyncio
22
import logging
33
import uuid
4-
from typing import Any, Awaitable, Coroutine, cast
4+
from collections.abc import Awaitable, Coroutine
5+
from typing import Any, cast
56

67
import aiodocker
78
from aiodocker.containers import DockerContainer
@@ -16,8 +17,8 @@ def _wrap_async_call(fct: Awaitable[Any]) -> Any:
1617

1718
def _nvidia_smi_docker_config(cmd: list[str]) -> dict[str, Any]:
1819
return {
19-
"Cmd": ["nvidia-smi"] + cmd,
20-
"Image": "nvidia/cuda:10.0-base",
20+
"Cmd": ["nvidia-smi", *cmd],
21+
"Image": "nvidia/cuda:12.2.0-base-ubuntu22.04",
2122
"AttachStdin": False,
2223
"AttachStdout": False,
2324
"AttachStderr": False,

0 commit comments

Comments
Ā (0)