This repo provides an up-to-date Dockerfile with the latest version of PyTorch and Nvidia Apex (for training accelerating).
docker pull jetrunner/pytorch-apex:release-x.x.x
The version x.x.x should be consistent with PyTorch version.
Currently supporting pytorch >= 1.2.0.
Content type
Image
Digest
Size
5.2 GB
Last updated
almost 6 years ago
docker pull davidezanella/pytorch-apex