Use nightly channel for pytorch, audio and vision to catch regression…

…s before a release
zjhzjh111 · Aug 31, 2021 · fb303a6 · fb303a6
2 parents e331265 + ee94dc2
commit fb303a6
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/scripts/install.sh b/scripts/install.sh
@@ -4,10 +4,7 @@ set -e
 . ~/miniconda3/etc/profile.d/conda.sh
 conda activate base
 
-# Install the latest release of pytorch and torchvision
-conda install -y pytorch torchvision -c pytorch
-# Also install torchaudio
-conda install -y -c pytorch torchaudio
+conda install -y pytorch torchvision torchaudio -c pytorch-nightly
 
 # Dependencies required to load models
 conda install -y regex pillow tqdm boto3 requests numpy\