mirror of
https://github.com/YuzuZensai/spleeter.git
synced 2026-01-06 04:32:43 +00:00
fix: gpu distribution
This commit is contained in:
13
setup.py
13
setup.py
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user