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?


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)


นี่คือสิ่งที่ฉันทำเพื่อให้ 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