Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
apex
Commits
b5a7c5f9
Commit
b5a7c5f9
authored
Jan 08, 2020
by
ptrblck
Committed by
mcarilli
Jan 08, 2020
Browse files
add WAR for pip>=19.3.1 (#652)
* add WAR for pip>=19.3.1 * remove pipmain, use extras_require instead
parent
0ce8ad3e
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
2 deletions
+3
-2
setup.py
setup.py
+3
-2
No files found.
setup.py
View file @
b5a7c5f9
...
...
@@ -2,7 +2,6 @@ import torch
from
setuptools
import
setup
,
find_packages
import
subprocess
from
pip._internal
import
main
as
pipmain
import
sys
import
warnings
import
os
...
...
@@ -31,10 +30,11 @@ if TORCH_MAJOR == 0 and TORCH_MINOR < 4:
cmdclass
=
{}
ext_modules
=
[]
extras
=
{}
if
"--pyprof"
in
sys
.
argv
:
with
open
(
'requirements.txt'
)
as
f
:
required_packages
=
f
.
read
().
splitlines
()
pipmain
([
"install"
]
+
required_packages
)
extras
[
'pyprof'
]
=
required_packages
try
:
sys
.
argv
.
remove
(
"--pyprof"
)
except
:
...
...
@@ -209,4 +209,5 @@ setup(
description
=
'PyTorch Extensions written by NVIDIA'
,
ext_modules
=
ext_modules
,
cmdclass
=
cmdclass
,
extras_require
=
extras
,
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment