This repository was archived by the owner on Sep 19, 2022. It is now read-only.
This repository was archived by the owner on Sep 19, 2022. It is now read-only.
Pytorch Docker image pytorch/pytorch:1.2-cuda10.0-cudnn7-runtime does not have cuda so unable to use GPU #245
Description
the pytorcg docker image pytorch/pytorch:1.0-cuda10.0-cudnn7-runtime used in examples/mnist Dockerfile cannot use GPU for mnist.py eaxmple always giving cuda as False.
This is because the cuda is not installed properly
kubectl exec -it pytorch-dist-mnist-gloo-worker-0 -n default -- /bin/bash
which cuda will give no results.
We have to use an appropriate Docker image which is capable of running the mnist usig GPU