Skip to content

GitLab

Unverified Commit 35e25914 authored by Jinjing Zhou's avatar Jinjing Zhou Committed by GitHub
Browse files

Update version check of PyTorch to 1.5 (#2079)

parent debfd4bf
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/dgl/backend/pytorch/tensor.py
View file @ 35e25914
......@@ -13,8 +13,8 @@ from ..._deprecate import kernel as K
from ...function.base import TargetCode
from ...base import dgl_warning
if LooseVersion(th.__version__) < LooseVersion("1.2.0"):
dgl_warning("Detected an old version of PyTorch. Suggest using torch>=1.2.0 "
if LooseVersion(th.__version__) < LooseVersion("1.5.0"):
dgl_warning("Detected an old version of PyTorch. Suggest using torch>=1.5.0 "
"for the best experience.")
def data_type_dict():
......
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