Unverified Commit 917a7552 authored by Jinjing Zhou's avatar Jinjing Zhou Committed by GitHub
Browse files

[Fix] fix version check (#2544)

parent 9a9a06eb
......@@ -17,7 +17,7 @@ PyTorch backend
---------------
Export ``DGLBACKEND`` as ``pytorch`` to specify PyTorch backend. The required PyTorch
version is 1.1.0 or later. See `pytorch.org <https://pytorch.org>`_ for installation instructions.
version is 1.5.0 or later. See `pytorch.org <https://pytorch.org>`_ for installation instructions.
MXNet backend
-------------
......@@ -44,5 +44,5 @@ instructions. In addition, DGL will set ``TF_FORCE_GPU_ALLOW_GROWTH`` to ``true`
.. code:: bash
pip install "tensorflow>=2.2.0rc1" # when using tensorflow cpu version
pip install "tensorflow>=2.2.0" # when using tensorflow cpu version
......@@ -14,7 +14,7 @@ from ...function.base import TargetCode
from ...base import dgl_warning
if LooseVersion(th.__version__) < LooseVersion("1.5.0"):
dgl_warning("Detected an old version of PyTorch. Suggest using torch>=1.5.0 "
raise Exception("Detected an old version of PyTorch. Please update 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