Merge pull request #45 from probcomp/include-torch

Add PyTorch install to the Dockerfile

Author: bzinberg

Dockerfile

@@ -15,7 +15,7 @@ RUN             git config --global user.name "Gen User"
 RUN             git config --global user.email "[email protected]"
 
 RUN             virtualenv -p /usr/bin/python3 /venv
-RUN             . /venv/bin/activate && pip install jupyter jupytext matplotlib tensorflow
+RUN             . /venv/bin/activate && pip install jupyter jupytext matplotlib tensorflow torch
 
 RUN             wget https://julialang-s3.julialang.org/bin/linux/x64/1.3/julia-1.3.1-linux-x86_64.tar.gz
 RUN             tar -xzv < julia-1.3.1-linux-x86_64.tar.gz