赞
踩
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")
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。