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