使用Tensorflow后端指定GPU ID

时间:2018-10-29 23:00:01

标签: python tensorflow keras gpu

我正在尝试使用GPU ID训练两种不同的模型。 我已经尝试过命令CUDA_VISIBLE_DEVICES=1 python filename.py,但是它使用的是GPU 0而不是1。 我还在代码中添加了os环境变量,但行为却相同。 我不确定如何解决此问题,因为这是第一次使用GPU。

0 个答案:

没有答案