Spaces:
Build error
Build error
File size: 521 Bytes
e703779 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
#!/bin/bash
# Set CUDA environment variables
export CUDA_HOME=/usr/local/cuda
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
# Create symlink if it doesn't exist
ln -sf /usr/local/cuda-12.4 /usr/local/cuda
# Install NVIDIA toolkit and dependencies
apt-get update && apt-get install -y \
cuda-toolkit \
ninja-build \
build-essential
# Install Python dependencies in the correct order
pip install torch==2.4.0 torchvision torchaudio
pip install -r requirements.txt |