ci: gpu testing on jenkins

This commit is contained in:
Jensun Ravichandran 2021-11-04 16:36:45 +01:00
parent f4519eb430
commit 75ab2897c4

38
Jenkinsfile vendored
View File

@ -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'
}
}
}
}
}