Ryuu
12/06/2023, 7:33 AMcustom_image = ImageSpec(
name="flyte-kfpytorch-plugin",
packages=["torch", "torchvision", "flytekitplugins-kfpytorch", "matplotlib", "tensorboardX",
registry="localhost:30000",
)
some lib is very large such as torch like above, This may reach timeout in pip install when building the image. Formally, how to set timeout for pip in this caseKevin Su
12/06/2023, 9:02 AMKevin Su
12/06/2023, 9:02 AMRyuu
12/06/2023, 9:11 AMRyuu
12/06/2023, 9:13 AMRyuu
12/06/2023, 9:15 AMKevin Su
12/06/2023, 10:17 AMImageSpec(apt_packages=["git"], packages=["--timeout=1000 numpy"], registry="localhost:30000")
Ryuu
12/06/2023, 10:29 AMRyuu
12/06/2023, 10:30 AMKevin Su
12/06/2023, 10:33 AM