一、tensorflow-gpu和CUDA、CUDNN、python相对应的版本:
tensorflow-gpu的版本一定要和CUDA对应,要不会无法运行。下面为tensorflow官网,包含最全。
https://tensorflow.google.cn/install/source
二、tensorflow-gpu运行时发生的问题:
如果版本一切配置正常,还是出现以下问题:
ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory
执行以下命令:
sudo ldconfig /usr/local/cuda-10.x/lib64
其中cuda-10.x为你下载的cuda版本号或你设置的包名。
完美解决!