fix: gpu distribution

This commit is contained in:
Félix Voituret
2019-11-21 17:48:46 +01:00
parent 4667e6f634
commit 94eb0e1ee5
3 changed files with 47 additions and 16 deletions

View File

@@ -15,7 +15,6 @@ __license__ = 'MIT License'
# Default project values.
project_name = 'spleeter'
project_version = '1.4.4'
device_target = 'cpu'
tensorflow_dependency = 'tensorflow'
tensorflow_version = '1.14.0'
here = path.abspath(path.dirname(__file__))
@@ -23,18 +22,6 @@ readme_path = path.join(here, 'README.md')
with open(readme_path, 'r') as stream:
readme = stream.read()
# Check if GPU target is specified.
if '--target' in sys.argv:
target_index = sys.argv.index('--target') + 1
target = sys.argv[target_index].lower()
sys.argv.remove('--target')
sys.argv.pop(target_index)
# GPU target compatibility check.
if device_target == 'gpu':
project_name = '{}-gpu'.format(project_name)
tensorflow_dependency = 'tensorflow-gpu'
# Package setup entrypoint.
setup(
name=project_name,