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.