Try switch to uv on runpod

utensil · Oct 28, 2024 · 4477be0 · 4477be0
1 parent 087508a
commit 4477be0
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 4 deletions.
diff --git a/yard-rs/runpod-xp/justfile b/yard-rs/runpod-xp/justfile
@@ -18,7 +18,7 @@ kill:
 test-kill ID:
     RUNPOD_POD_ID={{ID}} python kill.py
 
-run CONFIG="demo.yml":
+run CONFIG="rust.yml":
     uv run run.py {{ justfile_directory() / "tasks" / CONFIG }}
 
 gpus:
@@ -27,3 +27,12 @@ gpus:
         --header 'content-type: application/json' \
         --url 'https://api.runpod.io/graphql?api_key={{ apikey }}' \
         --data '{"query": "query GpuTypes { gpuTypes { id displayName memoryInGb } }"}'| jq | tee gpus.json |jq ".data.gpuTypes[].id"
+
+[unix]
+prep-uv:
+    curl -LsSf https://astral.sh/uv/install.sh | sh
+
+
+[windows]
+prep-uv:
+    powershell -ExecutionPolicy ByPass -c "irm https://astral.sh/uv/install.ps1 | iex"
diff --git a/yard-rs/runpod-xp/scripts/rust.sh b/yard-rs/runpod-xp/scripts/rust.sh
@@ -19,6 +19,8 @@ curl --proto '=https' --tlsv1.2 -sSf https://just.systems/install.sh | bash -s -
 curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- --default-toolchain nightly -y
 . "$HOME/.cargo/env"
 
+
+
 mkdir -p /content/
 cd /content/
 
@@ -45,7 +47,9 @@ cd ../../
 just ci
 
 cd /content/native-land/yard-rs/runpod-xp
-pip install -r requirements-runpod.txt
-just kill
+time just prep-uv
+time just monit
+sleep 60
+time just kill
 
-sleep infinity
+sleep infinity
diff --git a/yard-rs/runpod-xp/tasks/demo.yml → yard-rs/runpod-xp/tasks/rust.yml b/yard-rs/runpod-xp/tasks/demo.yml → yard-rs/runpod-xp/tasks/rust.yml