4090训练MegEngine报错

#1

【版本、环境信息】

  • MegEngine 版本：1.13.3+cu111
  • GPU型号：RTX4090）
  • 系统环境：manylinux2014_
  • python版本: 3.8
    使用4090训练模型报出错误：
    RuntimeError: NVIDIA GeForce RTX 4090(gpu0) with CUDA capability sm_89 is not compatible with the current MegEngine installation. The current MegEngine install supports CUDA capabilities sm_61 sm_70 sm_75 sm_80 sm_86. If you want to use the NVIDIA GeForce RTX 4090(gpu0) with MegEngine, please check the instructions at https://github.com/MegEngine/MegEngine/blob/master/scripts/cmake-build/BUILD_README.md
#2

目前理解 cuda11.8 以及对应的 whl 包是支持 sm 89 的，如果可以的话，可以考虑试一下 cu118 版本的包。

#3

目前理解 cuda118 以及对应的 whl 包应该是支持 sm 89的，如果可以的话，可以考虑尝试采用 cu118 版本的 whl 包。