#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