thijsvanloef / starnet-docker

A containerized variant of the Starnet++ application

Hello There! 👋

starnet-docker's Issues

CUDA not working on Docker under linux / Unraid

Describe the bug
I created a new unraid template for this docker container - it works and wil process images using the CPU. However, when trying to make the docker make use of my installed GPU, it is not working. Here is the docker run:

docker run
  -d
  --name='starnet-docker'
  --net='bridge'
  -e TZ="America/New_York"
  -e HOST_OS="Unraid"
  -e HOST_HOSTNAME="UnRaidNas"
  -e HOST_CONTAINERNAME="starnet-docker"
  -e 'PARALLEL'='false'
  -e 'STRIDE'='128'
  -e 'NVIDIA_VISIBLE_DEVICES'='GPU-fed446a2-d4a9-6b55-39a0-851833e73e1b'
  -e 'NVIDIA_DRIVER_CAPABILITIES'='all'
  -l net.unraid.docker.managed=dockerman
  -v '/mnt/user/Astronomy/Starnet/input/':'/home/starnet/application/input/':'rw'
  -v '/mnt/user/Astronomy/Starnet/output/':'/home/starnet/application/output/':'rw'
  --runtime=nvidia 'thijsvanloef/starnet-docker'
  --gpus=all
233ae4385ba5931ad2c82a388256cb8c092ef550ec87dcdcc0b3b4face8ae7da

the --runtime=nvidia, and other nvidia parameter are necessary for passing in a GPU to a container on unraid. I verified the GPU was visible in the container while it was running by using the console and using nvidia-smi which showed my GPU as have no jobs or running processes.

I also noticed in the docker tha the container size was only about 600mb, whereas the github repo lists the CUDA size at over 2 Gb. Not sure, perhaps I am pulling the wrong version, but this is the only on on dockerhub.

To Reproduce
Steps to reproduce the behavior:

1. Created docker in unraid
2. Passed in required flags
   Expected behavior
   expected to see the gpu do the processing.

Screenshots
If applicable, add screenshots to help explain your problem.

Desktop (please complete the following information):

• OS: Unraid
  -nvidia Quadro P400 GPU