From ad01287f2f1f9f728f2c3e7a6c6ea323bee88e86 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 12 Dec 2023 14:14:38 -0800 Subject: [PATCH] test --- userbenchmark/release-test/setup_env.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/userbenchmark/release-test/setup_env.sh b/userbenchmark/release-test/setup_env.sh index 88ae1d1620..c3a1fbd355 100644 --- a/userbenchmark/release-test/setup_env.sh +++ b/userbenchmark/release-test/setup_env.sh @@ -38,7 +38,8 @@ conda install -y -c pytorch ${MAGMA_VERSION} # install pip version of pytorch and torchvision if [[ ${PYTORCH_CHANNEL} == "pytorch-test" ]]; then - pip3 install torch==2.1.2 torchvision --index-url https://download.pytorch.org/whl/test/cu${CUDA_VERSION//./} + pip3 install --force-reinstall torch==2.1.2 torchvision=0.16.2 torchaudio=2.1.2 --index-url https://download.pytorch.org/whl/test/cu${CUDA_VERSION//./} + # pip3 install --force-reinstall torch==2.1.2 torchvision else pip3 install --force-reinstall torch torchvision fi