Skip to content

GitLab

Unverified Commit 5640b129 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[CI] Upgrade software version of CI docker image (#4189) 


Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent 1dddaad4
Hide whitespace changes
Inline Side-by-side
Showing with 20 additions and 14 deletions
Jenkinsfile
View file @ 5640b129
......@@ -206,7 +206,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:cu101_v220217"
image "dgllib/dgl-ci-cpu:cu101_v220629"
args "-u root"
alwaysPull true
}
......@@ -224,7 +224,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-gpu:cu101_v220217"
image "dgllib/dgl-ci-gpu:cu101_v220629"
args "-u root"
alwaysPull true
}
......@@ -261,7 +261,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:cu101_v220217"
image "dgllib/dgl-ci-cpu:cu101_v220629"
alwaysPull true
}
}
......@@ -278,7 +278,7 @@ pipeline {
agent {
docker {
label "linux-gpu-node"
image "dgllib/dgl-ci-gpu:cu101_v220217"
image "dgllib/dgl-ci-gpu:cu101_v220629"
args "--runtime nvidia"
alwaysPull true
}
......@@ -307,7 +307,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:cu101_v220217"
image "dgllib/dgl-ci-cpu:cu101_v220629"
alwaysPull true
}
}
......@@ -328,7 +328,7 @@ pipeline {
agent {
docker {
label "linux-gpu-node"
image "dgllib/dgl-ci-gpu:cu101_v220217"
image "dgllib/dgl-ci-gpu:cu101_v220629"
args "--runtime nvidia"
alwaysPull true
}
......@@ -350,7 +350,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:cu101_v220217"
image "dgllib/dgl-ci-cpu:cu101_v220629"
args "--shm-size=4gb"
alwaysPull true
}
......@@ -407,7 +407,7 @@ pipeline {
agent {
docker {
label "linux-gpu-node"
image "dgllib/dgl-ci-gpu:cu101_v220217"
image "dgllib/dgl-ci-gpu:cu101_v220629"
args "--runtime nvidia --shm-size=8gb"
alwaysPull true
}
......@@ -435,7 +435,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:cu101_v220217"
image "dgllib/dgl-ci-cpu:cu101_v220629"
alwaysPull true
}
}
......@@ -461,7 +461,7 @@ pipeline {
agent {
docker {
label "linux-gpu-node"
image "dgllib/dgl-ci-gpu:cu101_v220217"
image "dgllib/dgl-ci-gpu:cu101_v220629"
args "--runtime nvidia"
alwaysPull true
}
......
docker/Dockerfile.ci_cpu
View file @ 5640b129
# CI docker CPU env
# Adapted from github.com/dmlc/tvm/docker/Dockerfile.ci_cpu
FROM ubuntu:16.04
FROM ubuntu:18.04
ENV TZ=US
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
RUN apt-get update --fix-missing
......
docker/Dockerfile.ci_gpu
View file @ 5640b129
# CI docker GPU env
FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu16.04
FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
ENV TZ=US
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
RUN apt-get update --fix-missing
......
docker/install/conda_env/tensorflow_cpu.yml
View file @ 5640b129
......@@ -3,7 +3,7 @@ dependencies:
- python=3.6.9
- pip
- pip:
- tensorflow==2.2.0
- tensorflow==2.3.0
- tfdlpack
- pytest
- nose
......
docker/install/conda_env/tensorflow_gpu.yml
View file @ 5640b129
......@@ -3,7 +3,7 @@ dependencies:
- python=3.6.9
- pip
- pip:
- tensorflow==2.2.0
- tensorflow==2.3.0
- tfdlpack-gpu
- pytest
- nose
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment