Post Go back to editing

MAX78000的AI训练环境问题

出现问题如下:

描述:WSL2下python版本为3.10.12,pytorch版本为2.1.0,cuda版本为11.8,torchaudio版本为2.1.0,torchmetrics版本为1.2.0,torchnet版本为0.0.4,torchvision版本为0.16.0,并且已经正确安装,环境中都能检查到,但是运行实例训练模型代码出现 ModuleNotFoundError: No module named 'torchmetrics.detection.map'此问题,但是一查发现torchmetrics是安装成功的,所以不知道是哪里出了问题,需要解决方法。请大佬帮忙解决。

Parents Reply
  • 你好,你的方法并没有解决我的问题,可以看到我上面的配置 torch 和 torchvision 是均安装完成的,还有 torchmetrics 这3者均是用的 pip install 来安装的,如果说是兼容性问题,那么我该去安装那个版本的 torchmetrics , 这是我安装的 torchmetrics  的包中内容,也确实没有发现 torchmetrics.detection.map ,希望能解决此问题

Children