Pytorch, Medical Imaging, Image Segmentation, Nnunet, Machine Learning.
recommend a GPU to make predictions as this will be substantially faster than the other options. However, inference times are typically still manageable on CPU and MPS (Apple M1/M2). If using a GPU, it should have at least 4 GB of available (unused) VRAM.” Example hardware configurations Example