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

[waisonlee]

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

[gohjiayi]

在 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.