Update setup.py
Browse files
setup.py
CHANGED
|
@@ -56,16 +56,16 @@ setup(
|
|
| 56 |
],
|
| 57 |
)
|
| 58 |
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
|
|
|
| 56 |
],
|
| 57 |
)
|
| 58 |
|
| 59 |
+
install_options = [
|
| 60 |
+
"--disable-pip-version-check",
|
| 61 |
+
"--no-cache-dir",
|
| 62 |
+
"--no-build-isolation",
|
| 63 |
+
"--config-settings", "--build-option=--cpp_ext",
|
| 64 |
+
"--config-settings", "--build-option=--cuda_ext"
|
| 65 |
+
]
|
| 66 |
|
| 67 |
+
subprocess.run(
|
| 68 |
+
["pip", "install", "-v"] + install_options + ["git+https://github.com/sandeshrajbhandari/apex.git#egg=apex"],
|
| 69 |
+
check=True,
|
| 70 |
+
capture_output=True
|
| 71 |
+
)
|