Unverified Commit 0de1de3d authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[dev] upgrade minimum required torch version to 2.0.0 (#7358)

parent 3afa105b
......@@ -12,8 +12,8 @@ from ... import ndarray as nd
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():
......
#!/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
......@@ -10,8 +10,8 @@ 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.
......@@ -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
}
......
@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
......
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