https://flyte.org logo
#flyte-deployment
Title
# flyte-deployment
u

Uria Franko

09/19/2023, 8:50 PM
Hey guys, another error when using ImageSpec, any idea what might cause it?
Copy code
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"],
)
k

Kevin Su

09/19/2023, 8:57 PM
try
Copy code
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
k

Ketan (kumare3)

09/20/2023, 2:11 PM
Kevin this seems like a bug
3 Views