nnUnet with CUDA Driver 10.1
satara opened this issue · 1 comments
satara commented
Does nnUnet still work with CUDA Driver 10.1? When I install nnUnet with an older pytorch version it seems to update pytorch and when I train, the GPU is not available.
When I use the latest version of pytorch I get a "device kernel image is invalid" error when training, which seems to be related to CUDA incompatibility.
Is there a right combination of python, pytorch and cuda for nnUnet to work with cuda drive 10.1?
satara commented
I upgraded to 10.2, so cannot test this anymore