Uria Franko
09/19/2023, 8:50 PMimage = ImageSpec(
name="flyte",
base_image="docker pull <http://ghcr.io/flyteorg/flytekit:py3.9-1.9.1|ghcr.io/flyteorg/flytekit:py3.9-1.9.1>",
registry="<http://example.dkr.ecr.eu-central-1.amazonaws.com|example.dkr.ecr.eu-central-1.amazonaws.com>",
python_version="3.9",
cuda="11.6.2",
packages=["torch"],
)
Kevin Su
09/19/2023, 8:57 PMimage = ImageSpec(
name="flyte",
registry="<http://example.dkr.ecr.eu-central-1.amazonaws.com|example.dkr.ecr.eu-central-1.amazonaws.com>",
python_version="3.9",
cuda="11.6.2",
packages=["torch", "flytekit"],
)
ubuntu20.04.cuda….
, so that’s why the new image doesn’t contain flytekitKetan (kumare3)