-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
38 lines (34 loc) · 1.21 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import platform
from setuptools import find_packages, setup
install_requires = ["pytest", "tqdm", "scikit-learn", "matplotlib"]
print(f"Platform: {platform.system()}")
if platform.system() == "Linux":
print(f"Installing PyTorch with GPU support for {platform.system()}")
# Install PyTorch with GPU support
install_requires.append("torch")
install_requires.append("torchvision")
install_requires.append("torchaudio")
elif platform.system() == "Darwin":
# Install PyTorch for Mac
install_requires.append("torch")
install_requires.append("torchvision")
install_requires.append("torchaudio")
else:
raise ValueError(f"Unsupported platform: {platform.system()}")
setup(
name="ptmelt",
version="0.1.1",
description="PyTorch Machine Learning Toolbox (PT-MELT)",
url="https://github.com/NREL/pt-melt",
author="Nicholas T. Wimer",
author_email="[email protected]",
license="BSD 3-Clause License",
packages=find_packages(),
install_requires=install_requires,
classifiers=[
"Development Status :: 1 - Planning",
"Intended Audience :: Science/Research",
# "Programming Language :: Python :: 3.11",
],
# python_requires=">=3.8, <3.12",
)