From 697c13553ec60842f8576fe74c307b0793f0dc2a Mon Sep 17 00:00:00 2001 From: Michael Wyatt Date: Mon, 12 Sep 2022 10:46:49 -0700 Subject: [PATCH] bump P40 tests to torch 1.8 --- .../workflows/{nv-torch12-p40.yml => nv-torch18-p40.yml} | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) rename .github/workflows/{nv-torch12-p40.yml => nv-torch18-p40.yml} (91%) diff --git a/.github/workflows/nv-torch12-p40.yml b/.github/workflows/nv-torch18-p40.yml similarity index 91% rename from .github/workflows/nv-torch12-p40.yml rename to .github/workflows/nv-torch18-p40.yml index 944ba3beb19de..c5159507b70de 100644 --- a/.github/workflows/nv-torch12-p40.yml +++ b/.github/workflows/nv-torch18-p40.yml @@ -1,4 +1,4 @@ -name: nv-torch12-p40 +name: nv-torch18-p40 on: push: @@ -31,7 +31,7 @@ jobs: nvcc --version pip install --upgrade pip pip uninstall --yes torch torchvision - pip install torch==1.2.0 torchvision==0.4.0 + pip3 install torch==1.8.2 torchvision==0.9.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu101 python -c "import torch; print('torch:', torch.__version__, torch)" python -c "import torch; print('CUDA available:', torch.cuda.is_available())" @@ -59,4 +59,4 @@ jobs: run: | if [[ -d ./torch-extensions ]]; then rm -rf ./torch-extensions; fi cd tests - TORCH_EXTENSIONS_DIR=./torch-extensions pytest --color=yes --durations=0 --forked --verbose -n 4 unit/ --torch_ver="1.2" --cuda_ver="10" + TORCH_EXTENSIONS_DIR=./torch-extensions pytest --color=yes --durations=0 --forked --verbose -n 4 unit/ --torch_ver="1.8" --cuda_ver="10"