apache / mxnet

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more
https://mxnet.apache.org
Apache License 2.0
20.78k stars 6.79k forks source link

RTX 3080TI support #20451

Closed seventheli closed 3 years ago

seventheli commented 3 years ago

Description

Both docker Release 21.06 and mxnet_cu112-2.0.0b20210714-py3-none-manylinux2014_x86_64.whl not work for RTX 3080TI, is any version work on 3080TI?

docker run --gpus all -it --rm --env NVIDIA_DISABLE_REQUIRE=1 nvcr.io/nvidia/mxnet:21.06-py3 docker: Error response from daemon: OCI runtime create failed: container_linux.go:380: starting container process caused: process_linux.go:545: container init caused: Running hook #0:: error running hook: exit status 1, stdout: , stderr: nvidia-container-cli: initialization error: driver error: failed to process request: unknown.

References

https://docs.nvidia.com/deeplearning/frameworks/mxnet-release-notes/rel_21-06.html#rel_21-06

github-actions[bot] commented 3 years ago

Welcome to Apache MXNet (incubating)! We are on a mission to democratize AI, and we are glad that you are contributing to it by opening this issue. Please make sure to include all the relevant context, and one of the @apache/mxnet-committers will be here shortly. If you are interested in contributing to our project, let us know! Also, be sure to check out our guide on contributing to MXNet and our development guides wiki.

TristonC commented 3 years ago

@seventheli Which driver did you install on your machine? It should work with RTX 3080Ti.

seventheli commented 3 years ago

the problem is from the windows 10 dev version is too old, cu112 work fine in ubuntu 20.04