search

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.77k stars 6.8k forks source link

Install mxnet using pip #20279

Closed cuongngm closed 3 years ago

cuongngm commented 3 years ago

Description

I try to install mxnet using pip:

pip install mxnet-cu112

in some server using cuda 11.1 11.2 11.3 and google colab (cuda 11.2) but all the error when importing mxnet.

Error Message

OSError: libnvrtc.so.11.2: cannot open shared object file: No such file or directory (in server with cuda 11.1)
OSError: libcusolver.so.11: cannot open shared object file: No such file or directory (in google colab - cuda 11.2)
OSError: libcudnn.so.8: cannot open shared object file: No such file or directory (in server with cuda 11.2)
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.

leezu commented 3 years ago

Please install the cudnn, cusolver and nvrtc libraries that are part of the NVIDIA CUDA libraries.

cuongngm commented 3 years ago

in gg colab, i installed mxnet-cu110 and this problem fixed. Same for my server, I solved it by installing mxnet-cu100.