Search code examples
pythoncudamxnetgoogle-colaboratory

libnvrtc.so.8.0 missing on Google Colab (needed by mxnet)

I installed mxnet using

!pip install mxnet-cu80

However when I try to import mxnet I get error complaining that libnvrtc.so.8.0 is missing. Searching for it by !find / -name 'libnvrtc.*' returns nothing. Searching by google reveals that NVRTC is part of cuda toolkit. I have tried with both python 2 and 3 runtimes. The GPU acceleration is enabled.

What can I do to make mxnet working?

Solution

  • To get the missing library run

    !apt install libnvrtc8.0

    After that !find / -name 'libnvrtc.*' returns

    /usr/lib/x86_64-linux-gnu/libnvrtc.so.8.0
/usr/lib/x86_64-linux-gnu/libnvrtc.so.8.0.61

    and mxnet is working with the python 3 runtime.