windows10安装tensorflow GPU

1. 第一种可行方法

安装 GPU 驱动软件

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

下载依赖包

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

依赖包下载地址

离线安装依赖包如下:

tensorflow-gpu、tensorflow-gpu-estimator、tensorboard、keras-applications、keras-preprocessing、absl-py、wrapt、six、google-pasta、numpy、protobuf、termcolor、grpcio、wheel、astor、gast、h5py、setuptools、werkzeug、markdown。

版本号的话直接下载最新的即可。

安装依赖包

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

2. 最简单的方法

安装 miniconda 后,直接输入命令

conda install -c anaconda tensorflow-gpu

3. 检测是否支持gpu

import tensorflow as tf
tf.test.is_built_with_gpu_support

返回True

4. 参考

李想前
Anaconda Cloud

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。