Update install.sh

MolCrafts · Dec 2, 2024 · 2982dda · 2982dda
1 parent 09993b2
commit 2982dda
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.devcontainer/features/molpot/install.sh b/.devcontainer/features/molpot/install.sh
@@ -5,8 +5,8 @@ echo "Activating feature 'pytorch'"
 
 which pip > /dev/null || (apt update && apt install python3-pip -y -qq)
 
-if [ -z "${MOLPOTENABLECUDA}"]; then
-    echo "Installing pytorch with CUDA support"
+if [ -z "${MOLPOTENABLECUDA}" ]; then
+    echo "Installing pytorch with ${MOLPOTENABLECUDA} support"
     pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/${PYTORCHCUDAVERSION}
     echo "Pytorch with CUDA support installed!"
 else