ci: add gpu Dockerfile to jenkinsfile

This commit is contained in:
Alexander Engelsberger 2021-11-04 17:08:08 +01:00 committed by GitHub
parent 5bc8c57490
commit 328e789c86
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

34
Jenkinsfile vendored
View File

@ -1,6 +1,23 @@
pipeline {
agent none
stages {
stage('GPU') {
agent {
dockerfile {
filename 'gpu.Dockerfile'
dir '.ci'
args '--gpus 1'
}
}
steps {
sh 'pip install -U pip --progress-bar off'
sh 'pip install .[all] --progress-bar off'
sh './tests/test_examples.sh examples --gpu'
}
}
stage('CPU') {
parallel {
stage('3.10') {
@ -81,20 +98,7 @@ pipeline {
}
}
stage('GPU') {
agent {
docker {
image 'nvcr.io/nvidia/pytorch:21.10-py3'
args '--gpus 1'
}
}
steps {
sh 'pip install -U pip --progress-bar off'
sh 'pip install .[all] --progress-bar off'
sh './tests/test_examples.sh examples --gpu'
}
}
}
}
}