Skip to content

Commit 18ee381

Browse files
mrwyattiijeffra
andauthored
Upgrade P40 tests to torch 1.8 (deepspeedai#2316)
Co-authored-by: Jeff Rasley <jerasley@microsoft.com>
1 parent 9595dff commit 18ee381

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.github/workflows/nv-torch12-p40.yml renamed to .github/workflows/nv-torch18-p40.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
name: nv-torch12-p40
1+
name: nv-torch18-p40
22

33
on:
44
push:
@@ -32,7 +32,7 @@ jobs:
3232
nvcc --version
3333
pip install --upgrade pip
3434
pip uninstall --yes torch torchvision
35-
pip install torch==1.2.0 torchvision==0.4.0
35+
pip install torch==1.8.2 torchvision==0.9.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu101
3636
python -c "import torch; print('torch:', torch.__version__, torch)"
3737
python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
3838
@@ -60,4 +60,4 @@ jobs:
6060
run: |
6161
if [[ -d ./torch-extensions ]]; then rm -rf ./torch-extensions; fi
6262
cd tests
63-
TORCH_EXTENSIONS_DIR=./torch-extensions pytest --color=yes --durations=0 --forked --verbose -n 4 unit/ --torch_ver="1.2" --cuda_ver="10"
63+
TORCH_EXTENSIONS_DIR=./torch-extensions pytest --color=yes --durations=0 --forked --verbose -n 4 unit/ --torch_ver="1.8" --cuda_ver="10"

0 commit comments

Comments
 (0)