requirements-gpu.txt

absl-py==0.12.0
aenum==3.1.11
aiohttp==3.7.4.post0
async-timeout==3.0.1
attrs==21.2.0
cachetools==4.2.2
certifi==2020.12.5
chardet==4.0.0
charset-normalizer==2.0.12
colorful==0.5.4
dataclass-csv==1.3.0
dataclasses-json==0.5.4
frozendict==2.3.4
fsspec==2021.7.0
future==0.18.2
google-auth==1.35.0
google-auth-oauthlib==0.4.6
grpcio==1.37.1
idna==2.10
iniconfig==1.1.1
iteration-utilities==0.11.0
Jinja2==3.1.1
joblib==1.0.1
kubeflow-training==0.1.3
kubernetes==17.17.0
Markdown==3.3.4
MarkupSafe==2.1.1
marshmallow==3.13.0
marshmallow-enum==1.5.1
multidict==5.1.0
mypy-extensions==0.4.3
numpy==1.22.3
oauthlib==3.2.0
packaging==21.0
parameterized==0.8.1
Pillow==8.3.2
Pint==0.17
pluggy==1.0.0
prettyprinter==0.18.0
protobuf==3.16.0
psutil==5.8.0
py==1.11.0
pyasn1==0.4.8
pyasn1-modules==0.2.8
pyDeprecate==0.3.1
Pygments==2.10.0
pyparsing==2.4.7
pytest==7.1.2
python-dateutil==2.8.2
python-dotenv==0.17.1
pytz==2021.1
PyYAML==5.4.1
requests==2.25.1
requests-oauthlib==1.3.1
retrying==1.3.3
rsa==4.7.2
schedule==1.1.0
scikit-learn==1.0.2
scipy==1.8.0
six==1.16.0
stringcase==1.2.0
table-logger==0.3.6
tensorboard==2.5.0
tensorboard-data-server==0.6.1
tensorboard-plugin-wit==1.8.0
threadpoolctl==2.1.0
tomli==2.0.1
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.9.0+cu102
torchmetrics==0.5.0
torchsummary==1.5.1
-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.10.0+cu102
tqdm==4.49.0
typing-inspect==0.7.1
typing_extensions==4.1.1
urllib3==1.26.5
websocket-client==1.2.0
Werkzeug==1.0.1
yarl==1.6.3