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

Profiling Tensorflow

Posted on mar. 25 septembre 2018 in Tutorial by Corey Adams
Tagged with Tensorflow, profiling, memory, comparisons

A walkthrough of many of the current techniques for profiling time and memory usage for tensorflow.

Continue reading