pynvrtc di Google Colab: libnvrtc.so tidak ditemukan

Di notebook Google Colab, saya memasang paket python yang menggunakan pynvrtc untuk mengkompilasi beberapa kode CUDA.

Saya mendapat kesalahan:

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

Saya menjalankan !find / -name 'libnvrtc.*' dan mendapatkan

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

Pertanyaan saya adalah: Apakah saya seharusnya memiliki libnvrtc.so di sistem Google Colab saya? Bagaimana cara berkeliling jika saya hanya punya libnvrtc.so.8.0?


person Po C.    schedule 23.05.2018    source sumber
comment
Apa perpustakaan Python yang Anda coba gunakan?   -  person Bob Smith    schedule 23.05.2018
comment
Apakah instalasi python Anda sebenarnya 64 bit?   -  person talonmies    schedule 23.05.2018
comment
Ya. Saya yakin Google Colab adalah 64 bit. Perpustakaannya adalah github.com/jonas-koehler/s2cnn   -  person Po C.    schedule 23.05.2018


Jawaban (1)


Inilah yang saya lakukan agar pynvrtc berfungsi. Buat tautan simbolik untuk 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

Termotivasi oleh Apakah mungkin memasang cupy di google colab?

person Po C.    schedule 23.05.2018