- ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory
安装tensorflow报ImportError: libcublas.so.9.0: cannot open shared object file的解决方法(2018年4月) -
TensorFlow 识别不到GPU / TensorFlow GPU support
通过代码
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
可以看到目前设备对TensorFlow的支持. 以如下为例, 系统目前没有识别出安装的1080Ti显卡, 因此只能看到CPU.
正常识别出显卡的效果如下:
没有识别出显卡的原因有很多, 最为常见的是 TensorFlow 版本和cuDNN, Cuda Toolkit的版本不匹配, 或是安装的时候没有将路径添加到系统路径中导致无法找到.
可以通过查询 经过测试的构建配置 / tested_build_configurations 来获取TensorFlow, cuDNN, Cuda Toolkit版本对应关系.