search

tusen-ai / simpledet

A Simple and Versatile Framework for Object Detection and Instance Recognition
Apache License 2.0
3.08k stars 486 forks source link

Support for newer version of MXNet, CUDA 11, and 30-series NVIDIA GPU #351

Closed kzhang0718 closed 3 years ago

kzhang0718 commented 3 years ago

Hi,

Just wondering if there's any existing plan for supporting latest version of MXNet? The MXNet version used by simpledet does not support 30-series NVIDIA GPU, which requires CUDA 11. Yet I could not successfully compile this version of MXNet with CUDA 11.

Thanks.

RogerChern commented 3 years ago

I will see to it during the new year holidays.

On Thu, Dec 31, 2020 at 12:12 PM kzhang0718 notifications@github.com wrote:

Hi,

Just wondering if there's any existing plan for supporting latest version of MXNet? The MXNet version used by simpledet does not support 30-series NVIDIA GPU, which requires CUDA 11. Yet I could not successfully compile this version of MXNet with CUDA 11.

Thanks.

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/TuSimple/simpledet/issues/351, or unsubscribe https://github.com/notifications/unsubscribe-auth/ABGODH2KWDC3CNKXIHQR6ETSXP2Z5ANCNFSM4VPAQMYA .

huangzehao commented 3 years ago

@kzhang0718 We provide a dockerfile now. You can use mxnet 1.6.0 with cuda11.1. As for mxnet 1.7.0 and the master branch of mxnet, there still some problems due to the API change. Please refer https://github.com/TuSimple/simpledet/issues/350.

kzhang0718 commented 3 years ago

Thanks! Will try it out.