pub_soft/dgl
3
0
Fork 0
mirror of https://github.com/dmlc/dgl.git synced 2026-06-03 19:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

[GraphBolt]support torch2.4&cuda12.4 (#7629)

Browse Source
This commit is contained in:
pyynb
2024-08-01 16:36:43 +08:00
committed by GitHub
parent 65f85b50bb
commit d7461cd10a
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
script/create_dev_conda_env.sh
View File

@@ -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"

2
script/dgl_dev.yml.template
View File

@@ -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