From aa413c897a628093adce4362de31af43d079dfb3 Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Mon, 8 Apr 2024 18:13:18 -0400 Subject: [PATCH] Update setup_env.sh --- userbenchmark/release-test/setup_env.sh | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/userbenchmark/release-test/setup_env.sh b/userbenchmark/release-test/setup_env.sh index 97be2445b4..61fb8b9005 100644 --- a/userbenchmark/release-test/setup_env.sh +++ b/userbenchmark/release-test/setup_env.sh @@ -44,13 +44,13 @@ conda install -y -c pytorch ${MAGMA_VERSION} # install pip version of pytorch and torchvision if [[ ${PYTORCH_CHANNEL} == "pytorch-test" ]]; then - pip3 install --pre torch==2.2.2 --extra-index-url https://download.pytorch.org/whl/test/cu121 - pip3 install --pre torchvision==0.17.2 torchaudio==2.2.2 --extra-index-url https://download.pytorch.org/whl/test/cu121 + pip3 install --pre torch==2.3.0 --extra-index-url https://download.pytorch.org/whl/test/cu121 + pip3 install --pre torchvision==0.18.0 torchaudio==2.3.0 --extra-index-url https://download.pytorch.org/whl/test/cu121 #conda install -y pytorch=2.1.2 torchvision=0.16.2 torchaudio=2.1.2 pytorch-cuda=12.1 -c pytorch-test -c nvidia #pip3 install --force-reinstall --no-cache-dir torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/test/cu${CUDA_VERSION//./} else pip3 install --force-reinstall --no-cache-dir torch torchvision torchaudio - #conda install -y pytorch=2.1.1 torchvision=0.16.1 torchaudio=2.1.1 pytorch-cuda=12.1 -c pytorch -c nvidia + #conda install -y pytorch=2.2.2 torchvision=0.17.2 torchaudio=2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia fi python -c 'import torch; print(torch.__version__); print(torch.version.git_version)'