Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "a971c598b59532671a271520227cfd2fd54b1cd0"
Unverified Commit 917a7552 authored by Jinjing Zhou's avatar Jinjing Zhou Committed by GitHub
Browse files

[Fix] fix version check (#2544)

parent 9a9a06eb
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
docs/source/install/backend.rst
View file @ 917a7552
...@@ -17,7 +17,7 @@ PyTorch backend ...@@ -17,7 +17,7 @@ PyTorch backend
--------------- ---------------
Export ``DGLBACKEND`` as ``pytorch`` to specify PyTorch backend. The required PyTorch 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 MXNet backend
------------- -------------
...@@ -44,5 +44,5 @@ instructions. In addition, DGL will set ``TF_FORCE_GPU_ALLOW_GROWTH`` to ``true` ...@@ -44,5 +44,5 @@ instructions. In addition, DGL will set ``TF_FORCE_GPU_ALLOW_GROWTH`` to ``true`
.. code:: bash .. code:: bash
pip install "tensorflow>=2.2.0rc1" # when using tensorflow cpu version pip install "tensorflow>=2.2.0" # when using tensorflow cpu version
python/dgl/backend/pytorch/tensor.py
View file @ 917a7552
...@@ -14,8 +14,8 @@ from ...function.base import TargetCode ...@@ -14,8 +14,8 @@ from ...function.base import TargetCode
from ...base import dgl_warning from ...base import dgl_warning
if LooseVersion(th.__version__) < LooseVersion("1.5.0"): 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.") "for the best experience.")
def data_type_dict(): def data_type_dict():
return {'float16' : th.float16, return {'float16' : th.float16,
......
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