Windows上的TensorFlow
我花了几个小时才修复windows上的TensorFlow安装问题,所以下面是摘要:
检查TensorFlow-gpu是否工作(使用此代码):
带tf.device('/gpu:0'):a=tf.常数([1.0,2.0,3.0,4.0,5.0,6.0],形状=[2,3],名称='a')b=tf.常数([1.0,2.0,3.0,4.0,5.0,6.0],形状=[3,2],名称='b')c=tf.matmul(a,b)tf.Session()作为sess:打印(sess.run(c))
要检查可用CPU或GPU的列表(使用此代码):
从tensorflow.python.client导入设备lib打印(device_lib.list_local_devices())如果tf.test.gpu_device_name():print('默认GPU设备:{}'.format(tf.test.GPU_Device_name()))其他:打印(“请安装GPU版本的TF”)
使用CUDA和cuDNN在Windows上安装Tensorflow GPU
指南概述
- 在计算机上安装Nvidia的卡以及驱动程序
- 下载并安装CUDA
- 下载并“安装”cuDNN
- 卸载Tensorflow,安装Tensorflow GPU
- 更新系统上的%PATH%
- 验证安装
指南完整详细信息
确保
- 您已经卸载了tensorflow,并且只安装了tensonflow-gpu以完成任务。
- 根据您的项目设置,从pip和conda环境中卸载tensorflow,并只安装tensorfflow-gpu
- 设置PATH变量后,请确保注销或重新启动系统。
希望对您有所帮助:)