pynvrtc в Google Colab: libnvrtc.so не найден

Question

В записной книжке Google Colab я устанавливаю пакет python, который использует pynvrtc для компиляции некоторых кодов CUDA.

Я получил ошибку:

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

Я запускаю !find / -name 'libnvrtc.*' иget

/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.

!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

По мотивам Можно ли установить Cupy на Google Colab?