Skip to content

Commit

Permalink
Merge branch 'master' into kczhou-graphbolt-seeds-sampling
Browse files Browse the repository at this point in the history
  • Loading branch information
Rhett-Ying committed Apr 28, 2024
2 parents e314c26 + 0de1de3 commit d91a360
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 8 deletions.
4 changes: 2 additions & 2 deletions python/dgl/backend/pytorch/tensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,8 @@
from ...function.base import TargetCode
from ...utils import version

if version.parse(th.__version__) < version.parse("1.13.0"):
raise RuntimeError("DGL requires PyTorch >= 1.13.0")
if version.parse(th.__version__) < version.parse("2.0.0"):
raise RuntimeError("DGL requires PyTorch >= 2.0.0")


def data_type_dict():
Expand Down
12 changes: 6 additions & 6 deletions script/create_dev_conda_env.sh
Original file line number Diff line number Diff line change
@@ -1,17 +1,17 @@
#!/bin/bash

readonly CUDA_VERSIONS="11.6,11.7,11.8,12.1"
readonly TORCH_VERSION="1.13.0"
readonly PYTHON_VERSION="3.8"
readonly CUDA_VERSIONS="11.7,11.8,12.1"
readonly TORCH_VERSION="2.0.0"
readonly PYTHON_VERSION="3.10"

usage() {
cat << EOF
usage: bash $0 OPTIONS
examples:
bash $0 -c
bash $0 -g 11.7
bash $0 -g 11.7 -p 3.8
bash $0 -g 11.7 -p 3.8 -t 1.13.0
bash $0 -g 11.7 -p 3.10
bash $0 -g 11.7 -p 3.10 -t 2.0.0
bash $0 -c -n dgl-dev-cpu
Create a developement environment for DGL developers.
Expand All @@ -29,7 +29,7 @@ OPTIONS:
-p Create dev environment based on specified python version.
-s Run silently which indicates always 'yes' for any confirmation.
-t Create dev environment based on specified PyTorch version such
as '1.13.0'.
as '2.0.0'.
EOF
}

Expand Down
4 changes: 4 additions & 0 deletions tests/scripts/build_dgl.bat
Original file line number Diff line number Diff line change
@@ -1,6 +1,10 @@
@ECHO OFF
SETLOCAL EnableDelayedExpansion

ECHO "Current user: %USERNAME%"

python --version

CALL "C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools\VC\Auxiliary\Build\vcvars64.bat"
CALL mkvirtualenv --system-site-packages %BUILD_TAG%
DEL /S /Q build
Expand Down

0 comments on commit d91a360

Please sign in to comment.