Hi team, I am facing issues with using pytorch wit...
# ask-metaflow
t
Hi team, I am facing issues with using pytorch with Azure Kubernetes. Attached a screenshot of the error that I am seeing. My hunch is that my cluster supports CPU only and conda is installing the GPU version of pytorch because of which its failing somewhere. This is the dependency that I am using @conda_base(libraries={“pytorch:pytorch”“2.1.1", “pytorch:torchvision”“0.16.1", “pytorch:cpuonly”“*”}, python=“3.8”) And this works just fine locally.