saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix install_requires properly

Browse source
This commit is contained in:
Soumith Chintala 2019-08-08 05:11:13 -07:00
parent 11d7e8d85b
commit 64069120e4
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
setup.py
View file

@ -266,7 +266,11 @@ try:
except Exception:
pass
# the list of runtime dependencies required by this built package
install_requires = []
if os.getenv('PYTORCH_BUILD_VERSION'):
install_requires += ['numpy']
assert os.getenv('PYTORCH_BUILD_NUMBER') is not None
build_number = int(os.getenv('PYTORCH_BUILD_NUMBER'))
version = os.getenv('PYTORCH_BUILD_VERSION')
@ -346,10 +350,8 @@ def build_deps():
# Building dependent libraries
################################################################################
# the list of runtime dependencies required by this built package
install_requires = []
if sys.version_info <= (2, 7):
if sys.version_info <= (3, 0):
install_requires += ['future']
missing_pydep = '''
@ -372,8 +374,6 @@ class build_ext(setuptools.command.build_ext.build_ext):
cmake_cache_vars = defaultdict(lambda: False, cmake.get_cmake_cache_variables())
if cmake_cache_vars['USE_NUMPY']:
report('-- Building with NumPy bindings')
global install_requires
install_requires += ['numpy']
else:
report('-- NumPy not found')
if cmake_cache_vars['USE_CUDNN']: