Open ChaiBapchya opened 4 years ago
@astonzhang we need to check if there are two gpus to use x_shards[0] and x_shards[1]
Sure. Currently we require that the machine has to have at least 2 GPUs.
@ChaiBapchya For now, can you put such notebooks that require > 2 GPUs in exclusions of config.ini: https://github.com/d2l-ai/d2l-en/blob/master/config.ini#L28 ?
And should this be for local testing or pushed as a change to master?
local testing
Upon running on latest master
Command
It fails with
IndexError: list index out of range
Stack Trace