<#380 Add H200 partitions> Pull request opened by ...
# flyte-github
a
#380 Add H200 partitions Pull request opened by davidmirror-ops Adding the supported MIG profiles for the NVIDIA H200 device. Tested like this
Copy code
acc_env = flyte.TaskEnvironment(
    name="mig",
    resources=flyte.Resources(gpu=flyte.GPU(device="H200", quantity=1, partition="2g.35gb")),
    image=custom_image,
 
)
flyteorg/flyte-sdk