Deleted torchaudio and torchvision for faster preparation

NAG-DevOps · Oct 3, 2024 · c5abfe0 · c5abfe0
1 parent d085b25
commit c5abfe0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/jupyterlabs/setup_conda.sh b/src/jupyterlabs/setup_conda.sh
@@ -8,5 +8,5 @@ setenv CONDA_PKGS_DIRS /speed-scratch/$USER/Jupyter/pkgs
 conda create -p /speed-scratch/$USER/Jupyter/jupyter-env -y
 conda activate /speed-scratch/$USER/Jupyter/jupyter-env
 conda install -c conda-forge jupyterlab -y
-pip3 install --quiet torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+pip3 install --quiet torch --index-url https://download.pytorch.org/whl/cu118
 exit