From 18ee381eb344be9c504a09bd3b3a03c0199d595f Mon Sep 17 00:00:00 2001 From: Michael Wyatt Date: Mon, 12 Sep 2022 15:49:46 -0700 Subject: [PATCH] Upgrade P40 tests to torch 1.8 (#2316) Co-authored-by: Jeff Rasley --- .../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 29b9f891c3bb..6e57243bc31f 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: @@ -32,7 +32,7 @@ jobs: nvcc --version pip install --upgrade pip pip uninstall --yes torch torchvision - pip install torch==1.2.0 torchvision==0.4.0 + pip 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())" @@ -60,4 +60,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"