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包,在安装tensor...
原创
2019-08-23 11:24:55 ·
9573 阅读 ·
3 评论