Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "118c5be94a2b8eb90fa41a2ceb59b3a8de9e0218"
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 { ...@@ -318,7 +318,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-cpu-node" 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" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -545,7 +545,7 @@ pipeline { ...@@ -545,7 +545,7 @@ pipeline {
agent { agent {
docker { docker {
label "linux-gpu-node" 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" args "--runtime nvidia --shm-size=8gb"
alwaysPull true 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