Upload setup.sh with huggingface_hub
Browse files
setup.sh
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
# GPU Training Setup — Run this first on your Vast.ai instance
|
| 3 |
+
# Usage: bash setup.sh
|
| 4 |
+
|
| 5 |
+
set -e
|
| 6 |
+
|
| 7 |
+
echo "=== Setting up Humanoid SAC Training ==="
|
| 8 |
+
|
| 9 |
+
# Install dependencies
|
| 10 |
+
pip install torch gymnasium[mujoco] stable-baselines3 mujoco numpy
|
| 11 |
+
|
| 12 |
+
echo "=== Setup complete ==="
|
| 13 |
+
echo "Now run: python train_humanoid_gpu.py"
|