nnUNet
nnUNet copied to clipboard
about v1 how to inference in cpu?
I have understood how the nnunetv2 framework uses cpu inference, which is very convenient. But it seems that the nnunetv1 code can't set the device directly, do you know how to use cpu reasoning in the model inference phase and measure the inference time?
https://github.com/MIC-DKFZ/nnUNet/tree/nnunetv1?tab=readme-ov-file#installation
nnunetv1 probably doesn't support CPU.
HI, sorry for the late reply. We do not provide further support for V1. And as @gaojh135 already mentioned, V1 did not support that that back then. I am sorry, pleas switch to V2 Best, Constantin