感谢这篇博文:

【零基础保姆级教程】mmcv安装教程

python -c 'import torch;print(torch.__version__);print(torch.version.cuda)'

2.5.1即torch版本,cu121即cuda版本,下一行的12.1是相同意思,即cuda版本。

Installation — mmcv 2.2.0 documentation

链接中找到对应自己的版本。

      running build_ext
      C:\Python\Python310\lib\site-packages\torch\utils\cpp_extension.py:382: UserWarning: Error checking compiler version for cl: [WinError 2] 系统找不到指定的文件。
        warnings.warn(f'Error checking compiler version for {compiler}: {error}')
      building 'mmcv._ext' extension
      error: Microsoft Visual C++ 14.0 or greater is required. Get it with "Microsoft C++ Build Tools": https://visualstudio.microsoft.com/visual-cpp-build-tools/
      [end of output]


 

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐