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

fix(components/pytorch): Custom package installation - torchserve (#6055) 

* Swiching on custom installation

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

* Adding sklearn and captum

Signed-off-by: Shrinath Suresh <shrinath@ideas2it.com>
This commit is contained in:
shrinath-suresh 2021-07-20 11:52:15 +05:30 committed by GitHub
parent ee663d9593
commit 71317711be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
samples/contrib/pytorch-samples/bert/config.properties
View File

@ -4,7 +4,7 @@ metrics_address=http://0.0.0.0:8082
enable_metrics_api=true
metrics_format=prometheus
number_of_netty_threads=4
install_py_dep_per_model=false
install_py_dep_per_model=true
job_queue_size=10
service_envelope=kfserving
model_store=/mnt/models/model-store

3
samples/contrib/pytorch-samples/bert/requirements.txt
View File

@ -1,3 +1,4 @@
pytorch-lightning==1.3.7
pytorch-lightning
sklearn
captum
torchtext

2
samples/contrib/pytorch-samples/cifar10/config.properties
View File

@ -3,7 +3,7 @@ management_address=http://0.0.0.0:8081
metrics_address=http://0.0.0.0:8082
enable_metrics_api=true
metrics_format=prometheus
install_py_dep_per_model=false
install_py_dep_per_model=true
number_of_netty_threads=4
job_queue_size=10
service_envelope=kfserving

5
samples/contrib/pytorch-samples/cifar10/requirements.txt
View File

@ -1 +1,4 @@
pytorch-lightning==1.3.7
pytorch-lightning
sklearn
captum
torchtext