minor bug fix relevant for enabling GPU when installing
Browse files
setup.py
CHANGED
@@ -20,10 +20,11 @@ class InstallCommand(install):
|
|
20 |
self.cupy = "cupy"
|
21 |
|
22 |
def finalize_options(self):
|
23 |
-
print("
|
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 |
)
|