Sign inSign up

waggle/plugin-torch

By waggle

•Updated over 6 years ago

A base plugin for PyTorch with Torchvision

Image
0

791

waggle/plugin-torch repository overview

A base plugin for PyTorch 1.4.0 with torchvision 0.5.0.

For ARMv7 architecture, it is compiled with the following flags and does not support CUDA.

USE_CUDA=0
USE_MKLDNN=0
USE_NNPACK=0
USE_QNNPACK=0
USE_DISTRIBUTED=0

Tag summary

Content type

Image

Digest

Size

371.5 MB

Last updated

over 6 years ago

docker pull waggle/plugin-torch:1.4.0