update linux install targets

ref baa6b4dc36
This commit is contained in:
Alex "mcmonkey" Goodwin
2024-09-01 07:06:34 -07:00
parent 742dd3712b
commit ea231b14bb

View File

@@ -49,9 +49,9 @@ fi
# Install PyTorch based on GPU type
if [ "$GPU_TYPE" == "nv" ]; then
python3 -s -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121
python3 -s -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu124
elif [ "$GPU_TYPE" == "amd" ]; then
python3 -s -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0
python3 -s -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.1
fi
python3 -s -m pip install -r requirements.txt