[ci] replace the dngc ocker image with self-built pytorch image (#672)

pull/614/head^2
Frank Lee 2022-04-06 14:10:17 +08:00 committed by GitHub
parent 03e1d35931
commit cc236916c6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 7 deletions

View File

@ -14,7 +14,7 @@ jobs:
contains( github.event.pull_request.labels.*.name, 'Run Build and Test')
runs-on: [self-hosted, gpu]
container:
image: nvcr.io/nvidia/pytorch:21.07-py3
image: frankleeeee/pytorch-cuda:1.10.1-11.3.0
options: --gpus all --rm --ipc=host -v /data/scratch/cifar-10:/data/scratch/cifar-10
timeout-minutes: 40
steps:
@ -32,6 +32,6 @@ jobs:
pip install -v --no-cache-dir .
- name: Unit Testing
run: |
pytest tests
PYTHONPATH=$PWD pytest tests
env:
DATA: /data/scratch/cifar-10

View File

@ -14,10 +14,10 @@ jobs:
# 20.10: PyTorch 1.7.0a0+7036e91 + Python 3.8
# 20.12: PyTorch 1.8.0a0+1606899 + Python 3.8
# 21.06: PyTorch 1.9.0a0+c3d40fd + Python 3.8
container: ["nvcr.io/nvidia/pytorch:20.07-py3",
"nvcr.io/nvidia/pytorch:20.10-py3",
"nvcr.io/nvidia/pytorch:20.12-py3",
"nvcr.io/nvidia/pytorch:21.06-py3"]
container: ["frankleeeee/pytorch-cuda:1.9.0-11.1.1",
"frankleeeee/pytorch-cuda:1.8.1-11.1.1",
"frankleeeee/pytorch-cuda:1.7.1-11.0.3",
"frankleeeee/pytorch-cuda:1.6.0-10.2"]
container:
image: ${{ matrix.container }}
options: --gpus all --rm --ipc=host -v /data/scratch/cifar-10:/data/scratch/cifar-10
@ -35,7 +35,7 @@ jobs:
pip install -v --no-cache-dir .
- name: Unit Testing
run: |
pytest tests
PYTHONPATH=$PWD pytest tests
env:
DATA: /data/scratch/cifar-10