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

pynvrtc on Google Colab: libnvrtc.so not found

在 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 怎么办?

以下是我为使 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

受到

的激励