S

SiftZendeskTest

Issue with CUDA availability in Flyte task

Summary

The user is experiencing an issue where torch.cuda.is_available() returns False within a Flyte task, but returns True when executing a command in the same pod using kubectl exec. They are seeking possible reasons for this discrepancy and mention using a custom Docker image based on python:3.10-slim with flytekit-1.13.5 installed.

Status
open
Tags
    Source
    #ask-the-community