Flyte enables production-grade orchestration for machine learning workflows and data processing created to accelerate local workflows to production.

Flyte

CleanShot 2023-09-19 at 23.49.04@2x.png

Hey guys,
another error when using ImageSpec, any idea what might cause it?
```
image = 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"],
)```

try
```image = 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"],
)```

when passing cuda, flytekit will replace base image with `ubuntu20.04.cuda….` , so that’s why the new image doesn’t contain flytekit