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

В блокноте 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

Мой вопрос: полагаю ли я иметь libnvrtc.so в моей системе Google Colab? Как я могу обойти, если у меня есть только libnvrtc.so.8.0?


python google-colaboratory cuda
person Po C.    schedule 23.05.2018    source источник
comment
Какую библиотеку Python вы пытаетесь использовать?   -  person Bob Smith    schedule 23.05.2018
comment
Ваша установка Python на самом деле 64-битная?   -  person talonmies    schedule 23.05.2018
comment
Да. Я считаю, что Google Colab 64-битный. Библиотека находится на github.com/jonas-koehler/s2cnn.   -  person Po C.    schedule 23.05.2018

Ответы (1)


arrow_upward
1
arrow_downward

Вот что я сделал, чтобы заставить 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?

person Po C.    schedule 23.05.2018