Google Colab中pynvrtc报libnvrtc.so缺失的问题咨询与解决求助

Google Colab中是否应该存在libnvrtc.so？ #

其实Colab的CUDA环境里，默认不一定会有不带版本号的libnvrtc.so文件。系统通常会通过版本化的库文件（比如你找到的libnvrtc.so.8.0、libnvrtc.so.8.0.61）来管理CUDA库，而无版本号的文件一般是作为符号链接指向当前默认的版本库。Colab可能没有自动创建这个符号链接，所以才会导致你的Python包找不到它。

仅拥有版本化libnvrtc.so.8.0时的解决方法 #

你只需要手动创建一个从版本化库到无版本号文件名的符号链接，就能让pynvrtc找到所需的库文件了，步骤如下：

1. 在Colab的代码单元格中执行以下命令，创建软链接：

   !sudo ln -s /usr/lib/x86_64-linux-gnu/libnvrtc.so.8.0 /usr/lib/x86_64-linux-gnu/libnvrtc.so
   

   这个命令会将现有的libnvrtc.so.8.0链接为libnvrtc.so，让依赖它的Python包可以正常找到。

2. （可选）如果创建链接后仍然报错，可以尝试临时设置LD_LIBRARY_PATH环境变量，确保系统能找到这个库文件：

   !export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH
   

   或者在运行你的Python代码前，在代码里设置环境变量：

   import os
   os.environ['LD_LIBRARY_PATH'] = '/usr/lib/x86_64-linux-gnu:' + os.environ.get('LD_LIBRARY_PATH', '')
   

完成这些步骤后，再重新安装或运行你的Python包，应该就不会再出现libnvrtc.so找不到的错误了。

内容的提问来源于stack exchange，提问作者Po C.