Skip to content

GitLab

Unverified Commit 665c3798 authored by Tingyu Wang's avatar Tingyu Wang Committed by GitHub
Browse files

update cugraph docker image for CI (#5614)

parent 1ab0170a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
Jenkinsfile
View file @ 665c3798
......@@ -318,7 +318,7 @@ pipeline {
agent {
docker {
label "linux-cpu-node"
image "rapidsai/cugraph_nightly_torch-cuda:11.5-base-ubuntu18.04-py3.9-pytorch1.12.0-rapids22.12"
image "rapidsai/cugraph_stable_torch-cuda:11.8-base-ubuntu20.04-py3.10-pytorch2.0.0-rapids23.04"
args "-u root"
alwaysPull true
}
......@@ -545,7 +545,7 @@ pipeline {
agent {
docker {
label "linux-gpu-node"
image "rapidsai/cugraph_nightly_torch-cuda:11.5-base-ubuntu18.04-py3.9-pytorch1.12.0-rapids22.12"
image "rapidsai/cugraph_stable_torch-cuda:11.8-base-ubuntu20.04-py3.10-pytorch2.0.0-rapids23.04"
args "--runtime nvidia --shm-size=8gb"
alwaysPull true
}
......
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