#3771 [Core feature] Imagespec to support GPU dependencies Issue created by zeryx Motivation: Why do you think this is important? Image spec does not support adding GPU dependencies. It's possible with envd to support cuda https://envd.tensorchord.ai/api/starlark/v1/install.html#cuda Goal: What should the final outcome look like, ideally? I should be able to define an imagespec file similar to the following: default = ImageSpec( name="gpu", base_image="ghcr.io/flyteorg/flytekit:py3.8-1.6.0", registry="ghcr.io/some_user", packages=["flytekit>=1.6.0", "bayesian-optimization==1.4.3"], gpu=["cuda:11.04-cudnn7"], python_version="3.9" ) Describe alternatives you've considered An alternative is to build a base image that contains the particular CUDA / ROCm / OpenCL dependency - however that requires building and maintaining Propose: Link/Inline OR Additional context No response Are you sure this issue hasn't been raised already? ☑︎ Yes Have you read the Code of Conduct? ☑︎ Yes flyteorg/flyte