amend

pytorch · Oct 9, 2023 · 8961a31 · 8961a31
1 parent 8c0ee17
commit 8961a31
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/.circleci/unittest/linux/scripts/install.sh b/.circleci/unittest/linux/scripts/install.sh
@@ -27,26 +27,26 @@ git submodule sync && git submodule update --init --recursive
 printf "Installing PyTorch with %s\n" "${CU_VERSION}"
 if [[ "$TORCH_VERSION" == "nightly" ]]; then
   if [ "${CU_VERSION:-}" == cpu ] ; then
-      pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
+      python3 -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
   else
-      pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/$CU_VERSION
+      python3 -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/$CU_VERSION
   fi
 elif [[ "$TORCH_VERSION" == "stable" ]]; then
     if [ "${CU_VERSION:-}" == cpu ] ; then
-      pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+      python3 -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
   else
-      pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/$CU_VERSION
+      python3 -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/$CU_VERSION
   fi
 else
   printf "Failed to install pytorch"
   exit 1
 fi
 
 printf "* Installing tensordict\n"
-python setup.py develop
+python3 setup.py develop
 
 # install torchsnapshot nightly
-pip3 install git+https://github.com/pytorch/torchsnapshot --no-build-isolation
+python3 -m pip install git+https://github.com/pytorch/torchsnapshot --no-build-isolation
 
 # smoke test
 python -c "import functorch;import torchsnapshot"