amend

pytorch · Dec 4, 2024 · 97722ea · 97722ea
1 parent 53ae51f
commit 97722ea
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/unittest/linux_libs/scripts_habitat/install.sh b/.github/unittest/linux_libs/scripts_habitat/install.sh
@@ -21,9 +21,9 @@ git submodule sync && git submodule update --init --recursive
 
 printf "Installing PyTorch with %s\n" "${CU_VERSION}"
 if [[ "$TORCH_VERSION" == "nightly" ]]; then
-  pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
+  pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu121 -U
 elif [[ "$TORCH_VERSION" == "stable" ]]; then
-  pip3 install torch --index-url https://download.pytorch.org/whl/cu121
+  pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu121
 fi
 
 # install tensordict

diff --git a/.github/unittest/linux_libs/scripts_habitat/setup_env.sh b/.github/unittest/linux_libs/scripts_habitat/setup_env.sh
@@ -72,4 +72,5 @@ git clone https://github.com/facebookresearch/habitat-lab.git
 cd habitat-lab
 pip3 install -e habitat-lab
 pip3 install -e habitat-baselines  # install habitat_baselines
+pip3 install "numpy<2.0"
 conda run python -m pip install "gym[atari,accept-rom-license]" pygame