kubeflow
/
pipelines
mirror of https://github.com/kubeflow/pipelines.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore(components/pytorch): pytorch samples build fixes (#6338) 

* build dependency fixes

Signed-off-by: Shrinath Suresh <shrinath@ideas2it.com>

* Pinning PTL to 1.4.2

Signed-off-by: Shrinath Suresh <shrinath@ideas2it.com>

* Revert "Pinning PTL to 1.4.2"

This reverts commit 8e207f0d3b.

Signed-off-by: Shrinath Suresh <shrinath@ideas2it.com>

* Pinning PTL to 1.4.2

Signed-off-by: Shrinath Suresh <shrinath@ideas2it.com>
This commit is contained in:
shrinath-suresh 2021-08-16 10:36:05 +05:30 committed by GitHub
parent 9bec83f098
commit 5e597b13c8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
components/PyTorch/pytorch-kfp-components/setup.py
View File

@ -24,10 +24,8 @@ from setuptools import setup, find_packages
def make_required_install_packages():
return [
"kfp>=1.6.1",
"pytorch-lightning>=1.4.0",
"torch>=1.7.1",
"torchserve>=0.3.0",
"torch-model-archiver",
]

1
components/PyTorch/pytorch-kfp-components/tox.ini
View File

@ -30,6 +30,7 @@ deps =
mock
pytorch-lightning
torchmetrics
kfp
depends =

6
samples/contrib/pytorch-samples/Dockerfile
View File

@ -15,18 +15,12 @@ ARG BASE_IMAGE=pytorch/pytorch:latest
FROM ${BASE_IMAGE}
RUN apt-get update && apt-get install -y git
COPY . .
RUN pip install --no-cache-dir -r requirements.txt
RUN pip install --no-cache-dir pytorch-kfp-components
RUN pip install --no-cache-dir pytorch-lightning
RUN pip install torchmetrics==0.4.1
ENV PYTHONPATH /workspace
ENTRYPOINT /bin/bash

7
samples/contrib/pytorch-samples/requirements.txt
View File

@ -1,7 +1,6 @@
boto3
image
matplotlib
torch
matplotlib==3.4.2
pyarrow
sklearn
transformers
@ -15,4 +14,6 @@ wget
torch-model-archiver
minio
kfp
torchserve
tensorboard
torchmetrics
pytorch-lightning==1.4.2