search

shenweichen / GraphEmbedding

Implementation and experiments of graph embedding algorithms.
MIT License
3.63k stars 990 forks source link

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory #56

Open waisonlee opened 3 years ago

waisonlee commented 3 years ago

请问不用cuda9.0行吗 用cuda10的话需要修改哪里的代码

gohjiayi commented 2 years ago

在 setup.py 里发现需要用 TF 1.4.0-1.9.0, 就只可以用CUDA 9。想用CUDA 10就必须升级Tensorflow,查看这链接。

Within setup.py the requirements for TensorFlow is 1.4.0 to 1.9.0. According to the build guide here we can only use CUDA 9. Upgrading to CUDA 10 is possible only when you upgrade TensorFlow.

For even more advanced version of CUDA, I would recommend you to downgrade your CUDA instead as TF 1 to 2 requires substantial code change. You can use this repository phohenecker/switch-cuda to switch CUDA when required.