Skip to content

Commit

Permalink
[CI] Unpin mpmath (#1997)
Browse files Browse the repository at this point in the history
  • Loading branch information
@vmoens
vmoens authored Mar 6, 2024
1 parent fe6c070 commit 73e9b3f
Show file tree
Hide file tree
Showing 26 changed files with 54 additions and 54 deletions.
4 changes: 2 additions & 2 deletions .github/unittest/linux/scripts/run_all.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -124,9 +124,9 @@ 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 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/$CU_VERSION
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/$CU_VERSION
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_distributed/scripts/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,9 @@ git submodule sync && git submodule update --init --recursive

if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/$CU_VERSION -U
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/$CU_VERSION -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_examples/scripts/run_all.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -150,9 +150,9 @@ 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 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/$CU_VERSION
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/$CU_VERSION
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_ataridqn/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch torchvision --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
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_brax/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_d4rl/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_envpool/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,9 @@ git submodule sync && git submodule update --init --recursive

printf "Installing PyTorch with cu121"
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
fi

# smoke test
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_gen-dgrl/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch torchvision --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
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
2 changes: 1 addition & 1 deletion .github/unittest/linux_libs/scripts_habitat/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ git submodule sync && git submodule update --init --recursive

printf "Installing PyTorch with %s\n" "${CU_VERSION}"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --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
fi
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_jumanji/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_minari/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_openx/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch torchvision --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
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_pettingzoo/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_rlhf/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_robohive/install_and_run_test.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -39,9 +39,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_roboset/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_sklearn/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_smacv2/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_vd4rl/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch torchvision --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
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
4 changes: 2 additions & 2 deletions .github/unittest/linux_libs/scripts_vmas/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,9 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with cu121"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
else
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if [ "${CU_VERSION:-}" == cpu ] ; then
Expand Down
2 changes: 1 addition & 1 deletion .github/unittest/linux_optdeps/scripts/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ version="$(python -c "print('.'.join(\"${CUDA_VERSION}\".split('.')[:2]))")"
git submodule sync && git submodule update --init --recursive

printf "Installing PyTorch with %s\n" "${CU_VERSION}"
pip3 install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/$CU_VERSION
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/$CU_VERSION

# install tensordict
if [[ "$RELEASE" == 0 ]]; then
Expand Down
6 changes: 3 additions & 3 deletions .github/unittest/windows_optdepts/scripts/install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -41,13 +41,13 @@ git submodule sync && git submodule update --init --recursive
printf "Installing PyTorch with %s\n" "${cudatoolkit}"
if [[ "$TORCH_VERSION" == "nightly" ]]; then
if $torch_cuda ; then
python -m pip install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu118
python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu118
else
python -m pip install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
fi
elif [[ "$TORCH_VERSION" == "stable" ]]; then
if $torch_cuda ; then
python -m pip install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu118
python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu118
else
python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
fi
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/benchmarks.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ jobs:
python-version: 3.8
- name: Setup Environment
run: |
python -m pip install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
python -m pip install git+https://github.com/pytorch/tensordict
python setup.py develop
python -m pip install pytest pytest-benchmark
Expand Down Expand Up @@ -91,7 +91,7 @@ jobs:
echo /usr/local/bin >> $GITHUB_PATH
- name: Setup Environment
run: |
python3 -m pip install 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu121 -U
python3 -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu121 -U
python3 -m pip install git+https://github.com/pytorch/tensordict
python3 setup.py develop
python3 -m pip install pytest pytest-benchmark
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/benchmarks_pr.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ jobs:
python-version: 3.8
- name: Setup Environment
run: |
python -m pip install 'mpmath<1.4' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
python -m pip install git+https://github.com/pytorch/tensordict
python setup.py develop
python -m pip install pytest pytest-benchmark
Expand Down Expand Up @@ -102,7 +102,7 @@ jobs:
echo /usr/local/bin >> $GITHUB_PATH
- name: Setup Environment
run: |
python3 -m pip install 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu121 -U
python3 -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu121 -U
python3 -m pip install git+https://github.com/pytorch/tensordict
python3 setup.py develop
python3 -m pip install pytest pytest-benchmark
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/docs.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ jobs:
git version
# 5. Install PyTorch
python3 -m pip install 'mpmath<1.4' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U --quiet --root-user-action=ignore
python3 -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U --quiet --root-user-action=ignore
# 6. Install tensordict
python3 -m pip install git+https://github.com/pytorch/tensordict.git --quiet --root-user-action=ignore
Expand Down
Loading

0 comments on commit 73e9b3f

Please sign in to comment.