Pinning data to GPU in Tensorflow and PyTorch

Posted on mar. 02 octobre 2018 in Tutorial by Laura Domine
Tagged with memory, tensorflow, pytorch

Sometimes you might want to keep ('pin') some data on GPU. We demonstrate how to do it in Tensorflow and PyTorch.


Continue reading

Writing your own CUDA kernel (Part 1)

Posted on mar. 02 octobre 2018 in Tutorial by Laura Domine
Tagged with cuda, tensorflow, pytorch

First part of a tutorial serie to write your own CUDA kernel and use it in Tensorflow or PyTorch.


Continue reading

Writing your own CUDA kernel (Part 2)

Posted on mar. 02 octobre 2018 in Tutorial by Laura Domine
Tagged with cuda, tensorflow, pytorch

Second part of a tutorial serie to write your own CUDA kernel and use it in Tensorflow or PyTorch.


Continue reading