From 0939b41613422c8b1fb2176e28c528b86727c13c Mon Sep 17 00:00:00 2001 From: jalencato Date: Mon, 17 Jun 2024 17:59:17 -0700 Subject: [PATCH 1/5] Update Dockerfile.local --- docker/local/Dockerfile.local | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/local/Dockerfile.local b/docker/local/Dockerfile.local index 1172831fa9..c2404cab47 100644 --- a/docker/local/Dockerfile.local +++ b/docker/local/Dockerfile.local @@ -34,7 +34,7 @@ RUN pip install \ h5py==3.11.0 \ networkx==3.1 \ psutil==5.9.8 \ - pyarrow==14.0.0 \ + pyarrow==16.0.1 \ pydantic==2.7.0 \ scikit-learn==1.4.2 \ scipy==1.13.0 \ From e943c7d6261ef9ccb94c84e4a021f10ee2a460b1 Mon Sep 17 00:00:00 2001 From: jalencato Date: Mon, 17 Jun 2024 18:46:01 -0700 Subject: [PATCH 2/5] Update Dockerfile.local --- docker/local/Dockerfile.local | 1 + 1 file changed, 1 insertion(+) diff --git a/docker/local/Dockerfile.local b/docker/local/Dockerfile.local index c2404cab47..4532ddc21c 100644 --- a/docker/local/Dockerfile.local +++ b/docker/local/Dockerfile.local @@ -35,6 +35,7 @@ RUN pip install \ networkx==3.1 \ psutil==5.9.8 \ pyarrow==16.0.1 \ + numpy==1.26.0 \ pydantic==2.7.0 \ scikit-learn==1.4.2 \ scipy==1.13.0 \ From e8dbe6575cc425256e6792e08195cfb412af3d1f Mon Sep 17 00:00:00 2001 From: jalencato Date: Thu, 20 Jun 2024 11:32:45 -0700 Subject: [PATCH 3/5] Update Dockerfile.local --- docker/local/Dockerfile.local | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/local/Dockerfile.local b/docker/local/Dockerfile.local index 4532ddc21c..12693b3484 100644 --- a/docker/local/Dockerfile.local +++ b/docker/local/Dockerfile.local @@ -27,6 +27,7 @@ RUN curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2 RUN python3.9 -m venv /opt/gs-venv ENV PATH="/opt/gs-venv/bin:$PATH" +# Please downgrade the numpy version when using torch <= 2.2 # Install GraphStorm dependencies RUN pip install \ boto3==1.34.89 \ @@ -35,7 +36,6 @@ RUN pip install \ networkx==3.1 \ psutil==5.9.8 \ pyarrow==16.0.1 \ - numpy==1.26.0 \ pydantic==2.7.0 \ scikit-learn==1.4.2 \ scipy==1.13.0 \ From 47a03e2ded38898b1c346f5350f46cb538f11d19 Mon Sep 17 00:00:00 2001 From: jalencato Date: Mon, 24 Jun 2024 13:04:28 -0700 Subject: [PATCH 4/5] Update Dockerfile.local --- docker/local/Dockerfile.local | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/local/Dockerfile.local b/docker/local/Dockerfile.local index 12693b3484..52e7e1560e 100644 --- a/docker/local/Dockerfile.local +++ b/docker/local/Dockerfile.local @@ -44,7 +44,7 @@ RUN pip install \ ARG DGL_VERSION=2.2.1 ARG DGL_CUDA_VERSION=121 ARG OGB_VERSION=1.3.6 -ARG TORCH_VERSION=2.1.2 +ARG TORCH_VERSION=2.3 ARG TRANSFORMERS_VERSION=4.28.1 FROM base as base-cpu From 1ea98044f66397064f4300c06dfd6b0801368492 Mon Sep 17 00:00:00 2001 From: jalencato Date: Fri, 28 Jun 2024 11:14:55 -0700 Subject: [PATCH 5/5] Update Dockerfile.local --- docker/local/Dockerfile.local | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/local/Dockerfile.local b/docker/local/Dockerfile.local index 52e7e1560e..6b5974d295 100644 --- a/docker/local/Dockerfile.local +++ b/docker/local/Dockerfile.local @@ -41,7 +41,7 @@ RUN pip install \ scipy==1.13.0 \ && rm -rf /root/.cache -ARG DGL_VERSION=2.2.1 +ARG DGL_VERSION=2.3.0 ARG DGL_CUDA_VERSION=121 ARG OGB_VERSION=1.3.6 ARG TORCH_VERSION=2.3