From 041a09320570ec692819041190c7b025f4d27348 Mon Sep 17 00:00:00 2001 From: atalman Date: Wed, 13 Dec 2023 12:55:59 -0800 Subject: [PATCH] Test --- userbenchmark/release-test/setup_env.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/userbenchmark/release-test/setup_env.sh b/userbenchmark/release-test/setup_env.sh index 774e52758c..3ecc1748ca 100644 --- a/userbenchmark/release-test/setup_env.sh +++ b/userbenchmark/release-test/setup_env.sh @@ -41,7 +41,7 @@ if [[ ${PYTORCH_CHANNEL} == "pytorch-test" ]]; then 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 fi python -c 'import torch; print(torch.__version__); print(torch.version.git_version)'