Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "d9d94e12f36141db1836cf08db29dca8518cb5ad"
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
Hide 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 ...@@ -13,8 +13,8 @@ from ..._deprecate import kernel as K
from ...function.base import TargetCode from ...function.base import TargetCode
from ...base import dgl_warning from ...base import dgl_warning
if LooseVersion(th.__version__) < LooseVersion("1.2.0"): if LooseVersion(th.__version__) < LooseVersion("1.5.0"):
dgl_warning("Detected an old version of PyTorch. Suggest using torch>=1.2.0 " dgl_warning("Detected an old version of PyTorch. Suggest using torch>=1.5.0 "
"for the best experience.") "for the best experience.")
def data_type_dict(): 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