<#1688 Add Cuda to ImageSpec> Pull request opened ...
# flyte-github
a
#1688 Add Cuda to ImageSpec Pull request opened by pingsutw TL;DR
Copy code
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 flyteorg/flytekit All checks have passed 30/30 successful checks
#1688 Add Cuda to ImageSpec Pull request ready for review by pingsutw flyteorg/flytekit