Bump base dstack image version to 0.5 (#1541)

docs/docs/dev-environments.md

@@ -19,7 +19,7 @@ name: vscode
 
 python: "3.11"
 # Uncomment to use a custom Docker image
-#image: dstackai/base:py3.10-0.4-cuda-12.1
+#image: dstackai/base:py3.10-0.5-cuda-12.1
 
 ide: vscode
 

docs/docs/quickstart.md

@@ -37,7 +37,7 @@ Your folder can be a regular local folder or a Git repo.
 
     python: "3.11"
     # Uncomment to use a custom Docker image
-    #image: dstackai/base:py3.10-0.4-cuda-12.1
+    #image: dstackai/base:py3.10-0.5-cuda-12.1
 
     ide: vscode
 
@@ -96,7 +96,7 @@ Your folder can be a regular local folder or a Git repo.
 
     python: "3.11"
     # Uncomment to use a custom Docker image
-    #image: dstackai/base:py3.10-0.4-cuda-12.1
+    #image: dstackai/base:py3.10-0.5-cuda-12.1
 
     # Commands of the task
     commands:
@@ -170,7 +170,7 @@ Your folder can be a regular local folder or a Git repo.
 
     python: "3.11"
     # Uncomment to use a custom Docker image
-    #image: dstackai/base:py3.10-0.4-cuda-12.1
+    #image: dstackai/base:py3.10-0.5-cuda-12.1
 
     # Commands of the service
     commands:

docs/docs/reference/dstack.yml/service.md

@@ -68,7 +68,7 @@ port: 8000
     name: http-server-service
 
     # Any custom Docker image
-    image: dstackai/base:py3.10-0.4-cuda-12.1
+    image: dstackai/base:py3.10-0.5-cuda-12.1
 
     # Commands of the service
     commands:
@@ -89,7 +89,7 @@ port: 8000
     name: http-server-service
     
     # Any private Docker iamge
-    image: dstackai/base:py3.10-0.4-cuda-12.1
+    image: dstackai/base:py3.10-0.5-cuda-12.1
     # Credentials of the private registry
     registry_auth:
       username: peterschmidt85

docs/docs/reference/dstack.yml/task.md

@@ -92,7 +92,7 @@ type: dev-environment
 name: train    
 
 # Any custom Docker image
-image: dstackai/base:py3.10-0.4-cuda-12.1
+image: dstackai/base:py3.10-0.5-cuda-12.1
 
 # Commands of the task
 commands:
@@ -111,7 +111,7 @@ commands:
     name: train
     
     # Any private Docker image
-    image: dstackai/base:py3.10-0.4-cuda-12.1
+    image: dstackai/base:py3.10-0.5-cuda-12.1
     # Credentials of the private Docker registry
     registry_auth:
       username: peterschmidt85

examples/.dstack.yml

@@ -4,7 +4,7 @@ name: vscode
 
 python: "3.11"
 # Uncomment to use a custom Docker image
-#image: dstackai/base:py3.10-0.4-cuda-12.1
+#image: dstackai/base:py3.10-0.5-cuda-12.1
 
 ide: vscode
 

runner/cmd/shim/main.go

@@ -253,7 +253,7 @@ func getGpuInfo() [][]string {
 			"run",
 			"--rm",
 			"--gpus", "all",
-			"dstackai/base:py3.12-0.4-cuda-12.1",
+			"dstackai/base:py3.12-0.5-cuda-12.1",
 			"nvidia-smi", "--query-gpu=gpu_name,memory.total", "--format=csv",
 		},
 		StreamStdio: false,

src/dstack/version.py

@@ -1,3 +1,3 @@
 __version__ = None
 __is_release__ = False
-base_image = "0.4"
+base_image = "0.5"

src/tests/_internal/server/background/tasks/test_process_running_jobs.py

@@ -229,7 +229,7 @@ async def test_provisioning_shim(self, test_db, session: AsyncSession):
             ShimClientMock.return_value.submit.assert_called_once_with(
                 username="",
                 password="",
-                image_name="dstackai/base:py3.11-0.4-cuda-12.1",
+                image_name="dstackai/base:py3.11-0.5-cuda-12.1",
                 container_name="test-run-0-0",
                 container_user="root",
                 shm_size=None,

src/tests/_internal/server/routers/test_runs.py

@@ -147,7 +147,7 @@ def get_dev_env_run_plan_dict(
                     ],
                     "env": {},
                     "home_dir": "/root",
-                    "image_name": "dstackai/base:py3.8-0.4-cuda-12.1",
+                    "image_name": "dstackai/base:py3.8-0.5-cuda-12.1",
                     "job_name": f"{run_name}-0-0",
                     "replica_num": 0,
                     "job_num": 0,
@@ -283,7 +283,7 @@ def get_dev_env_run_dict(
                     ],
                     "env": {},
                     "home_dir": "/root",
-                    "image_name": "dstackai/base:py3.8-0.4-cuda-12.1",
+                    "image_name": "dstackai/base:py3.8-0.5-cuda-12.1",
                     "job_name": f"{run_name}-0-0",
                     "replica_num": 0,
                     "job_num": 0,