mlcube icon indicating copy to clipboard operation
mlcube copied to clipboard

mlcube using singularity run --nv

Open jeffcbecker opened this issue 3 years ago • 0 comments

Hi. I'm trying to run the MNIST example on an Nvidia V100, so the singularity run command needs the --nv argument to pull in CUDA and other Nvidia stuff. How do I do that with mlcube? Currently when I run with the singularity runner, it complains and just uses the CPU (although the execution is correct). Thanks.

jeffcbecker avatar Apr 27 '22 23:04 jeffcbecker