docker update

Author: manavmahan

pod-start.py

@@ -0,0 +1,38 @@
+import runpod
+import re
+import os
+
+API_KEY = os.getenv("RUNPOD_API_KEY")
+runpod.api_key = API_KEY
+
+# Replace with the actual template ID
+TEMPLATE_ID = "oqfjo3fqxu"
+
+# Fetch running instances
+running_pods = runpod.get_pods()
+manav_nums = []
+
+# Extract existing numbers from instance names
+for pod in running_pods:
+    match = re.match(r"manav-(\d+)", pod["name"])
+    if match:
+        manav_nums.append(int(match.group(1)))
+
+# Determine the next available number
+num = max(manav_nums) + 1 if manav_nums else 1
+print(f"Starting instance with number: {num}")
+
+pod_config = {
+    "name": f"manav-{num}",
+    "image_name": "manav",
+    "template_id": TEMPLATE_ID,
+    "gpu_type_id": "NVIDIA A40",
+    "cloud_type": "SECURE",
+    # "region": "CA-MTL-1",
+    "network_volume_id": "fbrfv5xpc9",  # Attach external disk
+    "env": {},  # Add any required environment variables here
+    "docker_args": "",  # Any specific Docker arguments
+}
+
+pod = runpod.create_pod(**pod_config)
+print(pod)

runpod/Dockerfile

@@ -1,5 +1,5 @@
 # Base image
-FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-devel
+FROM pytorch/pytorch:2.6.0-cuda12.6-cudnn9-devel
 
 # Update keys and repositories
 RUN apt-get update

runpod/requirements.txt

@@ -1,4 +1,5 @@
 beartype
+black
 classifier-free-guidance-pytorch
 einops
 einx[torch]

runpod/start.sh

@@ -85,7 +85,7 @@ start_jupyter() {
 #                               Main Program                                   #
 # ---------------------------------------------------------------------------- #
 
-start_nginx
+# start_nginx
 
 execute_script "/pre_start.sh" "Running pre-start script..."
 
@@ -95,9 +95,6 @@ setup_ssh
 start_jupyter
 export_env_vars
 
-mkdir /tmp/ramdisk
-mount -t tmpfs -o size=4G tmpfs /tmp/ramdisk
-
 execute_script "/post_start.sh" "Running post-start script..."
 
 echo "Start script(s) finished, pod is ready to use."