Google Colab 上的 pynvrtc：libnvrtc.so 未找到

Question

在 Google Colab 笔记本中，我正在安装一个 python 包，它使用 pynvrtc 来编译一些 CUDA 代码。

我收到错误：

OSError: libnvrtc.so: cannot open shared object file: No such file or directory

我运行!find / -name 'libnvrtc.*'然后得到

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

我的问题是：我的 Google Colab 系统中应该有 libnvrtc.so 吗？如果我只有 libnvrtc.so.8.0 怎么办？

Answer 1

以下是我为使 pynvrtc 工作所做的工作。为 libnvrtc.so.

创建符号 link

!apt -y install libnvrtc8.0
!ln -snf /usr/lib/x86_64-linux-gnu/libnvrtc.so.8.0 /usr/lib/x86_64-linux-gnu/libnvrtc.so

受到

的激励