3 ms·
Torch7 has amazing GPU support. If you wanna send a tensor (array) to the GPU, you just type array:cuda(). All operations you do from now on will be on the GPU.
by tabacof 12y ago
Torch7 has amazing GPU support. If you wanna send a tensor (array) to the GPU, you just type array:cuda(). All operations you do from now on will be on the GPU.
More involved example using a neural network: http://code.cogbits.com/wiki/doku.php?id=tutorial_cuda http://code.cogbits.com/wiki/doku.php?id=tutorial_cuda