search

dmlc / gluon-nlp

NLP made easy
https://nlp.gluon.ai/
Apache License 2.0
2.55k stars 538 forks source link

[Docker] CUDA11 Dockerfile support #1423

Open sxjscience opened 3 years ago

sxjscience commented 3 years ago

Description

Currently, our dockerfile supports cuda 10.2. We may try to upgrade it to CUDA11 to cope with A100: https://github.com/dmlc/gluon-nlp/tree/master/tools/docker