nnUNet2踩坑
环境配置
一开始我安装的pytorch1.13+cuda11.7,提示安装成功,下一步pip install nnunet2,结果不知道为什么这一步的一开始就会要求torch版本>2.0【from torch>=2.0.0->nnunetv2 】,然后我的torch包就自己升级了,升级之后就跟cuda不匹配,最后把环境删了重来
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia
后面的顺利进行,目前在跑第一折。