ci: gpu testing on jenkins
This commit is contained in:
parent
f4519eb430
commit
75ab2897c4
38
Jenkinsfile
vendored
38
Jenkinsfile
vendored
@ -3,12 +3,13 @@ pipeline {
|
||||
stages {
|
||||
stage('CPU') {
|
||||
parallel {
|
||||
stage('3.10'){
|
||||
stage('3.10') {
|
||||
agent {
|
||||
dockerfile {
|
||||
filename 'python310.Dockerfile'
|
||||
dir '.ci'
|
||||
}
|
||||
|
||||
}
|
||||
steps {
|
||||
sh 'pip install pip --upgrade --progress-bar off'
|
||||
@ -16,12 +17,14 @@ pipeline {
|
||||
sh './tests/test_examples.sh examples'
|
||||
}
|
||||
}
|
||||
stage('3.9'){
|
||||
|
||||
stage('3.9') {
|
||||
agent {
|
||||
dockerfile {
|
||||
filename 'python39.Dockerfile'
|
||||
dir '.ci'
|
||||
}
|
||||
|
||||
}
|
||||
steps {
|
||||
sh 'pip install pip --upgrade --progress-bar off'
|
||||
@ -29,12 +32,14 @@ pipeline {
|
||||
sh './tests/test_examples.sh examples'
|
||||
}
|
||||
}
|
||||
stage('3.8'){
|
||||
|
||||
stage('3.8') {
|
||||
agent {
|
||||
dockerfile {
|
||||
filename 'python38.Dockerfile'
|
||||
dir '.ci'
|
||||
}
|
||||
|
||||
}
|
||||
steps {
|
||||
sh 'pip install pip --upgrade --progress-bar off'
|
||||
@ -42,12 +47,14 @@ pipeline {
|
||||
sh './tests/test_examples.sh examples'
|
||||
}
|
||||
}
|
||||
stage('3.7'){
|
||||
|
||||
stage('3.7') {
|
||||
agent {
|
||||
dockerfile {
|
||||
filename 'python37.Dockerfile'
|
||||
dir '.ci'
|
||||
}
|
||||
|
||||
}
|
||||
steps {
|
||||
sh 'pip install pip --upgrade --progress-bar off'
|
||||
@ -55,12 +62,14 @@ pipeline {
|
||||
sh './tests/test_examples.sh examples'
|
||||
}
|
||||
}
|
||||
stage('3.6'){
|
||||
|
||||
stage('3.6') {
|
||||
agent {
|
||||
dockerfile {
|
||||
filename 'python36.Dockerfile'
|
||||
dir '.ci'
|
||||
}
|
||||
|
||||
}
|
||||
steps {
|
||||
sh 'pip install pip --upgrade --progress-bar off'
|
||||
@ -68,7 +77,24 @@ pipeline {
|
||||
sh './tests/test_examples.sh examples'
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
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'
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user