The Docker PyTorch image actually includes everything from PyTorch dependencies (numpy pyyaml scipy ipython mkl) to the PyTorch package itself, which could be pretty large because we built the image against all CUDA architectures. The CPU version should take less space.
We can use conda install pytorch-cpu torchvision -c pytorch to install PyTorch > 0.2 with Python 2.7. To automate this process, we can use the -y option, so the command would look like: conda install -y pytorch-cpu torchvision -c pytorch