Skip to content

GitLab

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

[CI]upgrade pytorch to 1.13+cpu_cu116 (#5977)

parent 4b456507
Show whitespace changes
Inline Side-by-side
Showing with 17 additions and 15 deletions
Jenkinsfile
View file @ e6e54304
......@@ -287,7 +287,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:v230606"
image "dgllib/dgl-ci-cpu:v230711"
args "-u root"
alwaysPull true
}
......@@ -305,7 +305,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-gpu:cu102_v230606"
image "dgllib/dgl-ci-gpu:cu116_v230711"
args "-u root"
alwaysPull true
}
......@@ -360,7 +360,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:v230606"
image "dgllib/dgl-ci-cpu:v230711"
alwaysPull true
}
}
......@@ -377,7 +377,7 @@ pipeline {
agent {
docker {
label "linux-gpu-node"
image "dgllib/dgl-ci-gpu:cu102_v230606"
image "dgllib/dgl-ci-gpu:cu116_v230711"
args "--runtime nvidia"
alwaysPull true
}
......@@ -406,7 +406,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:v230606"
image "dgllib/dgl-ci-cpu:v230711"
alwaysPull true
}
}
......@@ -427,7 +427,7 @@ pipeline {
agent {
docker {
label "linux-gpu-node"
image "dgllib/dgl-ci-gpu:cu101_v230210"
image "dgllib/dgl-ci-gpu:cu116_v230711"
args "--runtime nvidia"
alwaysPull true
}
......@@ -437,6 +437,8 @@ pipeline {
steps {
unit_test_linux('tensorflow', 'gpu')
}
// Tensorflow does not support cuda 11.6 yet.
when { expression { false } }
}
}
post {
......@@ -449,7 +451,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:v230606"
image "dgllib/dgl-ci-cpu:v230711"
args "--shm-size=4gb"
alwaysPull true
}
......@@ -501,7 +503,7 @@ pipeline {
agent {
docker {
label "linux-gpu-node"
image "dgllib/dgl-ci-gpu:cu102_v230606"
image "dgllib/dgl-ci-gpu:cu116_v230711"
args "--runtime nvidia --shm-size=8gb"
alwaysPull true
}
......@@ -529,7 +531,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:v230606"
image "dgllib/dgl-ci-cpu:v230711"
args "--shm-size=4gb"
alwaysPull true
}
......@@ -575,7 +577,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "dgllib/dgl-ci-cpu:v230606"
image "dgllib/dgl-ci-cpu:v230711"
alwaysPull true
}
}
......
docker/Dockerfile.ci_gpu
View file @ e6e54304
# CI docker GPU env
FROM nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04
FROM nvidia/cuda:11.6.2-cudnn8-devel-ubuntu18.04
ENV TZ=US
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
......
docker/install/conda_env/torch_cpu.yml
View file @ e6e54304
name: pytorch-ci
dependencies:
- python=3.7.0
- python=3.8
- pip
- pip:
- --find-links https://download.pytorch.org/whl/torch_stable.html
......
docker/install/conda_env/torch_cpu_pip.txt
View file @ e6e54304
......@@ -17,7 +17,7 @@ rdflib
requests[security]==2.28
scikit-learn
scipy
torch==1.12.0+cpu
torch==1.13.0+cpu
torchdata
torchmetrics
tqdm
docker/install/conda_env/torch_gpu.yml
View file @ e6e54304
name: pytorch-ci
dependencies:
- python=3.7.0
- python=3.8
- pip
- pip:
- --find-links https://download.pytorch.org/whl/torch_stable.html
......
docker/install/conda_env/torch_gpu_pip.txt
View file @ e6e54304
......@@ -15,7 +15,7 @@ rdflib
requests[security]==2.28
scikit-learn
scipy
torch==1.12.0+cu102
torch==1.13.0+cu116
torchdata
torchmetrics
tqdm
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