Skip to content

Commit

Permalink
Update setup and README to use PyTorch 1.0 (#1655)
Browse files Browse the repository at this point in the history
  • Loading branch information
neerajprad authored and fritzo committed Dec 7, 2018
1 parent 2fc96ff commit f87e087
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 29 deletions.
7 changes: 5 additions & 2 deletions .travis.yml
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,11 @@ cache:

install:
- pip install -U pip
- pip install torch_nightly==1.0.0.dev20181206 -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
- pip install torchvision --no-dependencies
- if [[ $TRAVIS_PYTHON_VERSION == 2.7 ]]; then
pip install http://download.pytorch.org/whl/cpu/torch-1.0.0-cp27-cp27mu-linux_x86_64.whl;
else
pip install http://download.pytorch.org/whl/cpu/torch-1.0.0-cp36-cp36m-linux_x86_64.whl;
fi
- pip install .[test]
- pip freeze

Expand Down
22 changes: 1 addition & 21 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -28,20 +28,12 @@ For more information, check out our [blog post](http://eng.uber.com/pyro).

### Installing a stable Pyro release

First install [PyTorch](http://pytorch.org/).

Install via pip:
**Install using pip:**

**Python 2.7.\*:**
```sh
pip install pyro-ppl
```

**Python 3.5:**
```sh
pip3 install pyro-ppl
```

**Install from source:**
```sh
git clone [email protected]:uber/pyro.git
Expand All @@ -62,18 +54,6 @@ Make sure that the models come from the same release version of the [Pyro source

For recent features you can install Pyro from source.

To install a compatible version of PyTorch, use the PyTorch nightly
[build](https://pytorch.org/). We recommend pinning to the specific
nightly build below that has been well tested.

```sh
build_ver=1.0.0.dev20181127 # <---- a well-tested PyTorch build
pip install torch_nightly==${build_ver} -f \
https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
```

Finally install Pyro using pip or from source as shown below.

**Install using pip:**

```sh
Expand Down
8 changes: 2 additions & 6 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,8 +56,7 @@
'matplotlib>=1.3',
'observations>=0.1.4',
'pillow',
# TODO: uncomment on release; using torch-nightly build
# 'torchvision',
'torchvision',
'visdom>=0.1.4',
'pandas',
'seaborn',
Expand Down Expand Up @@ -85,8 +84,7 @@
'numpy>=1.7',
'opt_einsum>=2.3.0',
'six>=1.10.0',
# TODO: uncomment on release; using torch-nightly build
# 'torch>=0.4.1',
'torch>=1.0.0',
'tqdm>=4.28',
],
extras_require={
Expand All @@ -96,7 +94,6 @@
'pytest==3.7',
'pytest-cov',
'scipy>=0.19.0',
'ipython<=6.5.0', # https://github.com/jupyter/jupyter_console/issues/158
],
'profile': ['prettytable', 'pytest-benchmark', 'snakeviz'],
'dev': EXTRAS_REQUIRE + [
Expand All @@ -109,7 +106,6 @@
'pypandoc',
'pytest==3.7',
'pytest-xdist',
'ipython<=6.5.0', # https://github.com/jupyter/jupyter_console/issues/158
'scipy>=0.19.0',
'sphinx',
'sphinx_rtd_theme',
Expand Down

0 comments on commit f87e087

Please sign in to comment.