在Kaggle使用torcheval报错：No module named 'torcheval'

问题描述 #

在Kaggle平台执行以下代码时：

from torcheval.metrics.functional import binary_f1_score
metric = binary_f1_score

触发ModuleNotFoundError，提示No module named 'torcheval'。用户已尝试升级至最新版PyTorch，但问题未解决，其余PyTorch相关模块可正常调用。

错误栈完整信息：

ModuleNotFoundError                       Traceback (most recent call last)
/tmp/ipykernel_28/3857382439.py in <module>
----> 1 from torcheval.metrics.functional import binary_f1_score
      2 metric = binary_f1_score
      3 model_no_CV = unet_1D().to(device)
      4 optimizer =  torch.optim.Adam(unet_1D.parameters(), lr=0.001)
      5 epochs_num = 30

ModuleNotFoundError: No module named 'torcheval'

解决步骤 #

1. 明确模块归属：torcheval是PyTorch生态下的独立度量工具库，不属于PyTorch核心包范畴，因此升级PyTorch不会自动安装该模块。
2. 在Kaggle中安装torcheval：在代码单元格中执行以下命令完成安装：

   !pip install torcheval
   

3. 版本兼容适配：若当前PyTorch版本较低，需安装与之兼容的torcheval版本。可先通过以下代码查看PyTorch版本：

   import torch
   print(torch.__version__)
   

   例如针对PyTorch 1.13.x版本，需指定安装对应版本：

   !pip install torcheval==0.0.7
   

4. 验证安装结果：安装完成后重新执行导入代码，即可正常使用binary_f1_score。

内容的提问来源于stack exchange，提问作者Diana