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

PyTorch/LArCV Classification Example with Data Set (v0.1.0)

Posted on mar. 09 janvier 2018 in tutorial by Taritree
Tagged with resnet, pytorch, classification, example

An example of training a classification network on the 5-particle LArCV training data using pytorch.


Continue reading