medimm PyTorch Medical 3D Image Models Installation Make sure that you have installed torch compatible with your CUDA version. Then install via pip pip install medimm