Skip to content

GitLab

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

[GraphBolt] upgrade CUDA to 17 and update CI gpu image to torch2.0+cu118 (#6505)

parent 864fd4f7
Show whitespace changes
Inline Side-by-side
Showing with 19 additions and 20 deletions
CMakeLists.txt
View file @ 9e26c8e3
cmake_minimum_required(VERSION 3.8) cmake_minimum_required(VERSION 3.18)
######################################## ########################################
# Borrowed and adapted from TVM project # Borrowed and adapted from TVM project
######################################## ########################################
......
Jenkinsfile
View file @ 9e26c8e3
...@@ -317,7 +317,7 @@ pipeline { ...@@ -317,7 +317,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v230711" image "dgllib/dgl-ci-cpu:v231027_1900"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -336,7 +336,7 @@ pipeline { ...@@ -336,7 +336,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-gpu:cu116_v230711" image "dgllib/dgl-ci-gpu:cu116_v231027_1900"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -391,7 +391,7 @@ pipeline { ...@@ -391,7 +391,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v230711" image "dgllib/dgl-ci-cpu:v231027_1900"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
...@@ -410,7 +410,7 @@ pipeline { ...@@ -410,7 +410,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-gpu" label "dgl-ci-linux-gpu"
image "dgllib/dgl-ci-gpu:cu116_v230711" image "dgllib/dgl-ci-gpu:cu116_v231027_1900"
args "-u root --runtime nvidia" args "-u root --runtime nvidia"
alwaysPull true alwaysPull true
} }
...@@ -463,7 +463,7 @@ pipeline { ...@@ -463,7 +463,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-gpu" label "dgl-ci-linux-gpu"
image "dgllib/dgl-ci-gpu:cu116_v230711" image "dgllib/dgl-ci-gpu:cu116_v231027_1900"
args "-u root --runtime nvidia" args "-u root --runtime nvidia"
alwaysPull true alwaysPull true
} }
...@@ -488,7 +488,7 @@ pipeline { ...@@ -488,7 +488,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v230711" image "dgllib/dgl-ci-cpu:v231027_1900"
args "-u root --shm-size=4gb" args "-u root --shm-size=4gb"
alwaysPull true alwaysPull true
} }
...@@ -541,7 +541,7 @@ pipeline { ...@@ -541,7 +541,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-gpu" label "dgl-ci-linux-gpu"
image "dgllib/dgl-ci-gpu:cu116_v230711" image "dgllib/dgl-ci-gpu:cu116_v231027_1900"
args "-u root --runtime nvidia --shm-size=8gb" args "-u root --runtime nvidia --shm-size=8gb"
alwaysPull true alwaysPull true
} }
...@@ -570,7 +570,7 @@ pipeline { ...@@ -570,7 +570,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v230711" image "dgllib/dgl-ci-cpu:v231027_1900"
args "-u root --shm-size=4gb" args "-u root --shm-size=4gb"
alwaysPull true alwaysPull true
} }
...@@ -618,7 +618,7 @@ pipeline { ...@@ -618,7 +618,7 @@ pipeline {
agent { agent {
docker { docker {
label "dgl-ci-linux-cpu" label "dgl-ci-linux-cpu"
image "dgllib/dgl-ci-cpu:v230711" image "dgllib/dgl-ci-cpu:v231027_1900"
args "-u root" args "-u root"
alwaysPull true alwaysPull true
} }
......
docker/install/conda_env/mxnet_cpu.yml
View file @ 9e26c8e3
...@@ -7,7 +7,7 @@ dependencies: ...@@ -7,7 +7,7 @@ dependencies:
- pytest - pytest
- nose - nose
- numpy - numpy
- cython - cython==0.29
- scipy - scipy
- networkx - networkx
- matplotlib - matplotlib
......
docker/install/conda_env/mxnet_gpu.yml
View file @ 9e26c8e3
...@@ -7,7 +7,7 @@ dependencies: ...@@ -7,7 +7,7 @@ dependencies:
- pytest - pytest
- nose - nose
- numpy - numpy
- cython - cython==0.29
- scipy - scipy
- networkx - networkx
- matplotlib - matplotlib
......
docker/install/conda_env/tensorflow_cpu.yml
View file @ 9e26c8e3
...@@ -7,7 +7,7 @@ dependencies: ...@@ -7,7 +7,7 @@ dependencies:
- pytest - pytest
- nose - nose
- numpy - numpy
- cython - cython==0.29
- scipy - scipy
- networkx - networkx
- matplotlib - matplotlib
......
docker/install/conda_env/tensorflow_gpu.yml
View file @ 9e26c8e3
...@@ -7,7 +7,7 @@ dependencies: ...@@ -7,7 +7,7 @@ dependencies:
- pytest - pytest
- nose - nose
- numpy - numpy
- cython - cython==0.29
- scipy - scipy
- networkx - networkx
- matplotlib - matplotlib
......
docker/install/ubuntu_install_build.sh
View file @ 9e26c8e3
# install cmake 3.15, cmake>=3.12 is required for CUDA 10.1 # Install cmake with minimum required version.
version=3.15 version=3.18
build=5 build=0
mkdir ~/temp mkdir ~/temp
cd ~/temp cd ~/temp
wget https://cmake.org/files/v$version/cmake-$version.$build-Linux-x86_64.sh wget https://cmake.org/files/v$version/cmake-$version.$build-Linux-x86_64.sh
......
graphbolt/CMakeLists.txt
View file @ 9e26c8e3
cmake_minimum_required(VERSION 3.5) cmake_minimum_required(VERSION 3.18)
project(graphbolt C CXX) project(graphbolt C CXX)
set (CMAKE_CXX_STANDARD 17) set (CMAKE_CXX_STANDARD 17)
...@@ -57,9 +57,8 @@ target_include_directories(${LIB_GRAPHBOLT_NAME} PRIVATE ${BOLT_DIR} ...@@ -57,9 +57,8 @@ target_include_directories(${LIB_GRAPHBOLT_NAME} PRIVATE ${BOLT_DIR}
"../third_party/pcg/include") "../third_party/pcg/include")
target_link_libraries(${LIB_GRAPHBOLT_NAME} "${TORCH_LIBRARIES}") target_link_libraries(${LIB_GRAPHBOLT_NAME} "${TORCH_LIBRARIES}")
# TODO: upgrade to 17 for consistency with CXX standard once our linux CI supports it.
if(USE_CUDA) if(USE_CUDA)
set_target_properties(${LIB_GRAPHBOLT_NAME} PROPERTIES CUDA_STANDARD 14) set_target_properties(${LIB_GRAPHBOLT_NAME} PROPERTIES CUDA_STANDARD 17)
endif() endif()
# The Torch CMake configuration only sets up the path for the MKL library when # The Torch CMake configuration only sets up the path for the MKL library when
......
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