File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 1
- name : nv-torch12 -p40
1
+ name : nv-torch18 -p40
2
2
3
3
on :
4
4
push :
32
32
nvcc --version
33
33
pip install --upgrade pip
34
34
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
36
36
python -c "import torch; print('torch:', torch.__version__, torch)"
37
37
python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
38
38
60
60
run : |
61
61
if [[ -d ./torch-extensions ]]; then rm -rf ./torch-extensions; fi
62
62
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"
You can’t perform that action at this time.
0 commit comments