Update .travis.yml

.travis.yml

@@ -1,31 +1,16 @@
 # https://travis-ci.org/pytorch/pytorch
-language: python
-dist: trusty
-git:
-  submodules: false
-
-# This reportedly works around an issue downloading packages from pypi on
-# travis.  Consider removing this after the underlying issue is fixed.
-# https://github.com/travis-ci/travis-ci/issues/2389
-sudo: false
+sudo: required
 
-install:
-  - sudo apt-get update
-  - wget https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh;
-  - bash miniconda.sh -b -p $HOME/miniconda
-  - export PATH="$HOME/miniconda/bin:$PATH"
-  - hash -r
-  - conda config --set always_yes yes --set changeps1 no
-  - conda update -q conda
-  # Useful for debugging any issues with conda
-  - conda info -a
+language: python
 
-  - conda install pytorch
-  - git clone https://github.com/pytorch/vision.git && cd vision && pip install -v .
+services:
+  - docker
 
-  - conda create -q -n test-environment python=$TRAVIS_PYTHON_VERSION pytorch scipy -c pytorch
-  - source activate test-environment
+before_install:
+- docker pull pytorch/pytorch
+- docker run -d -p 127.0.0.1:80:4567 pytorch/pytorch /bin/sh -c "cd /root/pytorch; bundle exec foreman start;"
+- docker ps -a
+- docker run pytorch/pytorch /bin/sh -c "cd /root/pytorch; bundle exec rake test"
 
 script:
-  - python train.py
-
+  - python model.py