Keras: 搭建GPU训练环境

安装

首先下载Anaconda安装包安装。

https://repo.anaconda.com/archive/Anaconda3-2020.02-Windows-x86_64.exe

安装后在开始菜单使用Administrator方式打开Anaconda Powershell Prompt(点击右键->以管理员方式运行)，输入如下命令安装keras-gpu。如果连接失败可能需要配置代理。

conda install keras-gpu

测试

运行Anaconda的Jupyter Notebook, 使用如下代码测试gpu是否工作：

import numpy as np
from keras import backend as K
K.tensorflow_backend._get_available_gpus()

[OUT] ['/job:localhost/replica:0/task:0/device:GPU:0']