andreped commited on
Commit
b44b8e4
1 Parent(s): dac5ddb

minor bug fix relevant for enabling GPU when installing

Browse files
Files changed (1) hide show
  1. setup.py +3 -3
setup.py CHANGED
@@ -20,10 +20,11 @@ class InstallCommand(install):
20
  self.cupy = "cupy"
21
 
22
  def finalize_options(self):
23
- print("CuPY version selected is: ", self.cupy)
24
  install.finalize_options(self)
25
 
26
- def run(self):
 
27
  required.append(self.cupy)
28
  install.run(self)
29
 
@@ -55,5 +56,4 @@ setup(
55
  ],
56
  python_requires='>=3.6',
57
  cmdclass={'install': InstallCommand},
58
- #py_modules=['livermask.utils'],
59
  )
 
20
  self.cupy = "cupy"
21
 
22
  def finalize_options(self):
23
+ print("CuPy version selected is: ", self.cupy)
24
  install.finalize_options(self)
25
 
26
+ def run(self):
27
+ global cupy
28
  required.append(self.cupy)
29
  install.run(self)
30
 
 
56
  ],
57
  python_requires='>=3.6',
58
  cmdclass={'install': InstallCommand},
 
59
  )