在多卡机器上训练的时候,经常会需要指定tensorflow程序跑在固定的GPU上,于是就有了如下手段:
export CUDA_VISIBLE_DEVICES=2,3
指定我们的程序只能看到GPU中序号为2与3的设备,于是启动后就可以看到:
我们的训练程序进程号是27088,可以看到它只跑在了2与3显卡上。
在多卡机器上训练的时候,经常会需要指定tensorflow程序跑在固定的GPU上,于是就有了如下手段:
export CUDA_VISIBLE_DEVICES=2,3
指定我们的程序只能看到GPU中序号为2与3的设备,于是启动后就可以看到:
我们的训练程序进程号是27088,可以看到它只跑在了2与3显卡上。