shellcheck

flexflow · Oct 19, 2024 · bf6be8c · bf6be8c
1 parent d8355ca
commit bf6be8c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/flexflow-environment/install_pytorch.sh b/docker/flexflow-environment/install_pytorch.sh
@@ -19,9 +19,9 @@ install_pytorch() {
   echo "Attempting to install PyTorch with CUDA ${major}.${minor} support..."
 
   # Run dry-run first
-  if pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu${major}${minor} --dry-run; then
+  if pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu"${major}${minor}" --dry-run; then
     echo "Dry-run succeeded, proceeding with actual installation..."
-    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu${major}${minor}
+    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu"${major}${minor}"
     return 0
   else
     echo "Dry-run failed for CUDA ${major}.${minor}."