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"