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

Question

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

kzhang0718 opened this issue 4 years ago · 3 comments

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.

Answer 1 · 2020-12-31T07:00:43.000Z

I will see to it during the new year holidays.

…

On Thu, Dec 31, 2020 at 12:12 PM kzhang0718 ***@***.***> 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 <#351>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ABGODH2KWDC3CNKXIHQR6ETSXP2Z5ANCNFSM4VPAQMYA> .

Answer 2 · 2021-01-14T05:10:42.000Z

@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 #350.

Answer 3 · 2021-01-18T03:49:03.000Z

Thanks! Will try it out.