get pytorch docker at
docker pull abhishekkr/pytorch:0.2.0
at it's core provides 2 main features
-
an n-dimensional tensor, similar to numpy, can run on GPUs
-
automatic differentitation for building and training neural networks
-
has dynamic computational graph constrution