Add Cuda to ImageSpec

[pingsutw]

TL;DR

from flytekit import task, workflow
from flytekit.image_spec import ImageSpec

new_flytekit = "git+https://github.com/flyteorg/flytekit@8c7c43ad310ec74b4e8c06c2b4604b6cb9692f86"
image_spec = ImageSpec(base_image="ubuntu20.04", packages=[new_flytekit], cuda="11.2.2", cudnn="8",
                       registry="pingsutw", apt_packages=["git"], python_version="3.9",)

Type

• Bug Fix
• Feature
• Plugin

Are all requirements met?

• Code completed
• Smoke tested
• Unit tests added
• Code documentation added
• Any pending items have an associated Issue

Complete description

^^^

Tracking Issue

NA

Follow-up issue

NA