keras 调用 GPU（tensorflow-GPU）心得

keras 调用 GPU（tensorflow）

1.

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"             #选用GPU序号

2.问题

若无法调用，可以重新建立一个虚拟环境，安装cuda，安装与cuda版本对应的cudnn包，在安装tensorflow-gpu（不用直接安装tensorflow）和kears。完成后重启设备即可使用。

1）建立conda的虚拟环境，并进入虚拟环境。

2）查看支持的cuda版本

$ conda search cudatoolkit
# Name                       Version           Build  Channel             
cudatoolkit                      9.0      h13b8566_0  pkgs/main           
cudatoolkit                      9.2               0  pkgs/main           
cudatoolkit                 10.0.130               0  pkgs/main           
cudatoolkit                 10.1.168               0  pkgs/main

选择安装你需要的cuda版本

再接着安装与之对应的cudnn，tensorflow-gpu版本

在安装keras库

感觉你的安装顺序不对有可能会导致GPU不可用。