开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

咨询个问题,ModelScope训练模型能设置使用cpu核数么?

咨询个问题,ModelScope训练模型能设置使用cpu核数么?

展开
收起
真的很搞笑 2023-04-17 08:07:26 395 0
1 条回答
写回答
取消 提交回答
  • 从您提供的输出信息来看,似乎在设置 PyTorch 的线程数时没有生效,线程数依然为 64。这可能是因为您的操作系统、硬件或软件环境导致的。下面是一些建议,希望能帮助您解决问题:

    确保您的代码在主程序的开始部分设置了线程数。将线程数的设置放在所有 PyTorch 操作之前,以确保设置能够在整个程序中生效。 请确保环境变量设置正确。您可以使用以下代码检查环境变量设置: python Copy code import os

    print("OMP_NUM_THREADS:", os.environ.get("OMP_NUM_THREADS")) print("MKL_NUM_THREADS:", os.environ.get("MKL_NUM_THREADS")) 请检查您的代码中是否有其他地方可能影响到线程数设置的部分,例如其他模块、包或者脚本。 请检查您使用的 PyTorch 版本是否存在已知的线程数设置问题。您可以通过运行 print(torch.version) 来检查您当前的 PyTorch 版本。如果可能,请尝试升级到最新版本的 PyTorch。 如果您在 Jupyter Notebook、Colab 或其他交互式环境中运行代码,请尝试在独立的 Python 脚本中运行相同的代码,以排除环境相关的问题。 如果问题仍然存在,请提供更多关于您的环境(操作系统、硬件配置、Python 版本等)和代码的详细信息,以便我能更好地为您提供帮助。,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2023-04-18 18:03:31
    赞同 展开评论 打赏

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
RDS SQL Server CPU高使用率性能优化 立即下载