Giters

d2l-ai / d2l-en

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

Home Page:https://D2L.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PyTorch Colab of Chapter 13.5 is having an IndexError

gab-chen opened this issue · comments

commented

Chapter 13.5 Training on Multiple GPUs
13.5.5. Distributing Data
IndexError: list index out of range
image

commented

@AnirudhDagar Can you take a look when available?
Found that you adapted the code to PyTorch.

commented

Hi @gab-chen, on google colab, you can only use one GPU, that is the limit from Google. This section of the book will probably fail on colab until they add multi gpu support. You can try to test the code in an environment where you have multiple gpus, and should be just fine.