We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d7f1575 commit 4ecc58bCopy full SHA for 4ecc58b
āservices/dask-sidecar/src/simcore_service_dask_sidecar/utils.py
@@ -1,7 +1,8 @@
1
import asyncio
2
import logging
3
import uuid
4
-from typing import Any, Awaitable, Coroutine, cast
+from collections.abc import Awaitable, Coroutine
5
+from typing import Any, cast
6
7
import aiodocker
8
from aiodocker.containers import DockerContainer
@@ -16,8 +17,8 @@ def _wrap_async_call(fct: Awaitable[Any]) -> Any:
16
17
18
def _nvidia_smi_docker_config(cmd: list[str]) -> dict[str, Any]:
19
return {
- "Cmd": ["nvidia-smi"] + cmd,
20
- "Image": "nvidia/cuda:10.0-base",
+ "Cmd": ["nvidia-smi", *cmd],
21
+ "Image": "nvidia/cuda:12.2.0-base-ubuntu22.04",
22
"AttachStdin": False,
23
"AttachStdout": False,
24
"AttachStderr": False,
0 commit comments