scott/test2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

more stuff

Browse Source
This commit is contained in:
Scott Register
2025-07-27 08:19:42 -07:00
parent 7431954482
commit 9faaf4ed57
5 changed files with 21 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
runpod_setup.sh
View File

@@ -58,20 +58,11 @@ pip install -r requirements.txt
print_status "Installing video processing libraries..."
pip install decord ffmpeg-python
# Install CuPy for GPU acceleration of stereo validation
# Install CuPy for GPU acceleration (CUDA 12 is standard on modern RunPod)
print_status "Installing CuPy for GPU acceleration..."
# Auto-detect CUDA version and install appropriate CuPy
if command -v nvidia-smi &> /dev/null; then
CUDA_VERSION=$(nvidia-smi | grep "CUDA Version" | awk '{print $9}' | cut -d. -f1-2)
echo "Detected CUDA version: $CUDA_VERSION"
if [[ "$CUDA_VERSION" == "11."* ]]; then
pip install cupy-cuda11x>=12.0.0 && print_success "Installed CuPy for CUDA 11.x"
elif [[ "$CUDA_VERSION" == "12."* ]]; then
pip install cupy-cuda12x>=12.0.0 && print_success "Installed CuPy for CUDA 12.x"
else
print_error "Unknown CUDA version, skipping CuPy installation"
fi
print_status "Installing CuPy for CUDA 12.x (standard on RunPod)..."
pip install cupy-cuda12x>=12.0.0 && print_success "Installed CuPy for CUDA 12.x"
else
print_error "NVIDIA GPU not detected, skipping CuPy installation"
fi