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