Skip to content

GitLab

Commit dc55a996 authored by Cubbee's avatar Cubbee Committed by mcarilli
Browse files

Fix typo in setup.py error message on torch version check (#219)

parent 0a991543
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ dc55a996
...@@ -22,7 +22,7 @@ ext_modules = [] ...@@ -22,7 +22,7 @@ ext_modules = []
if "--cpp_ext" in sys.argv or "--cuda_ext" in sys.argv: if "--cpp_ext" in sys.argv or "--cuda_ext" in sys.argv:
if TORCH_MAJOR == 0: if TORCH_MAJOR == 0:
raise RuntimeError("--cpp_ext requires Pytorch 1.0 or later, " raise RuntimeError("--cpp_ext requires Pytorch 1.0 or later, "
"found torch.__version__ = {}".format(torch.__version)) "found torch.__version__ = {}".format(torch.__version__))
from torch.utils.cpp_extension import BuildExtension from torch.utils.cpp_extension import BuildExtension
cmdclass['build_ext'] = BuildExtension cmdclass['build_ext'] = BuildExtension
......
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