Changes

Jump to: navigation, search

DPS921/PyTorch: Convolutional Neural Networks

2 bytes added, 15:06, 30 November 2020
Data Parallelism
Then, you can copy all your tensors to the GPU:
mytensor = my_tensor.to(device)
However, PyTorch will only use one GPU by default. In order to run on multiple GPUs you need to use <code>DataParallel</code>:
56
edits

Navigation menu