Skip to content

PyTorch

Examples images from the FashionMnist dataset

PyTorch / datasets / dataloader / data transfer to GPU – I – properties of some torchvision datasets

For an old fan of Tensorflow2 it is somewhat satisfactory to notice that some TF2- problems also exist in analogous form in a PyTorch environment. Anyone who has worked with visual data knows that one needs to modify/augment and transform the image data and then load them from some storage under CPU control to the GPU’s VRAM during network training.… Read More »PyTorch / datasets / dataloader / data transfer to GPU – I – properties of some torchvision datasets

CUDA and cudnn with Pytorch and Tensorflow in one virtual Python environment on your Linux system

One of the problems I recently ran into was the coexistence of Tensorflow2 [TF2] and PyTorch in the very same virtual Python environment. I just wanted to make experiments to compare the performance of some Keras-based models with the TF2-backend on one side and, on the other side, with the PyTorch-backend. My trouble resulted from a mismatch of two CUDA/cudnn… Read More »CUDA and cudnn with Pytorch and Tensorflow in one virtual Python environment on your Linux system