mlcube
mlcube copied to clipboard
mlcube using singularity run --nv
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.