windows10安装tensorflow GPU

1.安装方法

安装 GPU 驱动软件

根据个人电脑配置,安装 NVIDIA® 软件,cuda和cuDnn, 并配置好路径。

下载依赖包

这步很关键,大多数安装失败就是因为依赖包下载不够导致系统反复寻找相关依赖包。

依赖包下载地址

离线安装依赖包如下:tensorflow-gpu、absl-py、wrapt、keras-applications、six、google-pasta、tensorflow-gpu-estimator、numpy、protobuf、termcolor、keras-preprocessing、grpcio、wheel、astor、tensorboard、gast、h5py、setuptools、werkzeug、markdown。版本号的话直接下载最新的即可。

安装依赖包

命令为:
pip install 本地目录+python离线包
安装顺序自己适当调整即可。

  1. 检测是否支持gpu
import tensorflow as tf
tf.test.is_built_with_gpu_support

返回True

2.参考

李想前

https://www.jianshu.com/p/9c739c046b0e

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论