python - Google Colab 上缺少 libnvrtc.so.8.0（mxnet 需要）

Question

我使用安装了 mxnet

!pip install mxnet-cu80

但是，当我尝试时，import mxnet我收到错误抱怨libnvrtc.so.8.0丢失。通过搜索它!find / -name 'libnvrtc.*'不会返回任何内容。谷歌搜索显示NVRTC是 cuda 工具包的一部分。我尝试过使用 python 2 和 3 运行时。GPU 加速已启用。

我该怎么做才能使 mxnet 正常工作？

Answer 1

运行缺少的库

!apt install libnvrtc8.0

之后!find / -name 'libnvrtc.*'返回

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

mxnet 正在使用 python 3 运行时。