From a773f06bb6774f75bb7d7a67c7be0016bbddcde2 Mon Sep 17 00:00:00 2001 From: Michael Wyatt Date: Mon, 12 Sep 2022 10:46:49 -0700 Subject: [PATCH 1/2] 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 29b9f891c3bb..226519976e8d 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 + 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())" @@ -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" From 6d91e32b692dbde5601c7b4f8da1f1f173003d50 Mon Sep 17 00:00:00 2001 From: Michael Wyatt Date: Mon, 12 Sep 2022 10:55:06 -0700 Subject: [PATCH 2/2] Update nv-torch18-p40.yml --- .github/workflows/nv-torch18-p40.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/nv-torch18-p40.yml b/.github/workflows/nv-torch18-p40.yml index 226519976e8d..6e57243bc31f 100644 --- a/.github/workflows/nv-torch18-p40.yml +++ b/.github/workflows/nv-torch18-p40.yml @@ -32,7 +32,7 @@ jobs: nvcc --version pip install --upgrade pip pip uninstall --yes torch torchvision - pip3 install torch==1.8.2 torchvision==0.9.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu101 + 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())"