当前位置:   article > 正文

tensorflow-gpu无法安装的问题

tensorflow-gpu无法安装的问题

1.首先要配好与自己显卡对应的cuda 还有cudnn,然后去查询自己需要下载tensorflow的版本

如果你的tensorflow-gpu安装不上,有可能是因为你的电脑上已经有一个cpu版本的tensorflow导致不兼容可以使用下面代码检查一下,看看输出几个路径,其实只需要把之前下载过的和tensorflow卸载下来,然后找到对于的源安装就可以了。

import tensorflow as tf
print(tf.test.is_gpu_available())

print(tf.__version__)  # 查看tensorflow版本
print(tf.__path__)     # 查看tensorflow安装路径

a = tf.test.is_built_with_cuda()  # 判断CUDA是否可以用
b = tf.test.is_gpu_available(
    cuda_only=False,
    min_cuda_compute_capability=None
)  # 判断GPU是否可以用

print(a) # 显示True表示CUDA可用
print(b) # 显示True表示GPU可用

# 查看驱动名称
if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/菜鸟追梦旅行/article/detail/181528
推荐阅读
相关标签
  

闽ICP备14008679号