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 |
)
|