CCI-Bonn / HD-GLIO

Automated deep-learning based brain tumor segmentation on MRI
Apache License 2.0
26 stars 7 forks source link

GPU not recognized? #5

Open keithcdodd opened 3 years ago

keithcdodd commented 3 years ago

I use a Dell XPS 13 9300 with the GPU "Intel Corporation Iris Plus Graphics G7 (rev 07)". When I try to run HD-GLIO it gives an error "RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx". Admittedly, I get the same issue with HD-Bet unless I make it use my CPU.

Is there a way to get it to recognize and use my GPU? Or does it have to be NVIDIA based?

Hoping there is a way for it to work, as it is a GPU, so I would think it could be used. It is just not NVIDIA specifically.

Thanks!

FabianIsensee commented 3 years ago

Hi, HD-GLIO uses pytorch and pytorch needs an Nvidia GPU. There is unfortunately no way around that right now. Best, Fabian

keithcdodd commented 3 years ago

Gotcha. I tried a potential workaround by uninstalling pytorch and then reinstalling just for CPU (https://pytorch.org/get-started/locally/ --> select CPU). I then uninstalled and reinstalled HDGlio.

Now when I run I get the error "Segmentation fault". I am wondering if this would be an error because of using cpu only pytorch (and whether or not this attempt of a workaround is possible), or perhaps an issue with my data instead.

Thanks

FabianIsensee commented 3 years ago

Hi, it's unlikely to be related to your data. If you want me to double check you can send me one image (all modalities) and I will try it out myself