jetrunner/pytorch-apex repository overview

⁠pytorch-apex-docker

Docker Hub⁠

⁠Introduction

This repo provides an up-to-date Dockerfile with the latest version of PyTorch⁠ and Nvidia Apex⁠ (for training accelerating).

⁠Usage

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.