From d7461cd10a91ab24dacdc88d07168dfd855f1b0c Mon Sep 17 00:00:00 2001
From: pyynb <52124938+pyynb@users.noreply.github.com>
Date: Thu, 1 Aug 2024 16:36:43 +0800
Subject: [PATCH] [GraphBolt]support torch2.4&cuda12.4 (#7629)

---
 script/create_dev_conda_env.sh | 2 +-
 script/dgl_dev.yml.template    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/script/create_dev_conda_env.sh b/script/create_dev_conda_env.sh
index 01a8795f63..891be05954 100644
--- a/script/create_dev_conda_env.sh
+++ b/script/create_dev_conda_env.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 
-readonly CUDA_VERSIONS="11.8,12.1"
+readonly CUDA_VERSIONS="11.8,12.1,12.4"
 readonly TORCH_VERSION="2.1.0"
 readonly PYTHON_VERSION="3.10"
 
diff --git a/script/dgl_dev.yml.template b/script/dgl_dev.yml.template
index b41d4a3db8..708df84aa1 100644
--- a/script/dgl_dev.yml.template
+++ b/script/dgl_dev.yml.template
@@ -10,7 +10,7 @@ dependencies:
   - pandoc
   - pygraphviz
   - pip:
-    - --find-links https://download.pytorch.org/whl/torch_stable.html
+    - --find-links https://download.pytorch.org/whl/torch/
     - cmake>=3.18
     - cython
     - filelock