ImportError: libcublas.so.8.0: cannot open shared object file: No such file or directory

cesc-park / attend2u

🖼️ Attend to You: Personalized Image Captioning with Context Sequence Memory Networks. In CVPR, 2017. Expanded : Towards Personalized Image Captioning via Multimodal Memory Networks. In IEEE TPAMI, 2018.

MIT License

209 stars 57 forks source link

ImportError: libcublas.so.8.0: cannot open shared object file: No such file or directory #13

Closed robinsdeepak closed 4 years ago

robinsdeepak commented 5 years ago

Hey, I'm getting error while training the data after extraction, Can you help me with it? Capture

ecustWallace commented 4 years ago

I guess it's because your cudatoolkit is 9. or 10.. If you use anaconda, you can simply: conda install cudatoolkit==8.0 or some other version.

Or instead, you can use an updated tensorflow like conda install tensorflow-gpu==1.12.0rc2.

riven314 commented 4 years ago

@ecustWallace Thanks for the address I saw from requirements.txt that the source code requires tensorflow-gpu version 1.1.0 to run. Does tensorflow-gpu version 1.12 actually work for this source code?