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.