increase torch version to 1.0.0

Author: ozancaglayan

NEWS.md

@@ -1,6 +1,7 @@
 ## Release Notes
 
-This release supports Pytorch >= 0.4.1 including the recent 1.0 release.
+This release supports Pytorch >= 0.4.1 including the recent 1.0 release. The relevant
+`setup.py` and `environment.yml` files will default to `1.0.0` installation.
 
 ### v4.0.0 (18/12/2018)
  - **Critical**: `NumpyDataset` now returns tensors of shape `HxW, N, C` for 3D/4D convolutional features, `1, N, C` for 2D feature files. Models should be adjusted to adapt to this new shaping.

README.md

@@ -41,7 +41,7 @@ If you use **nmtpytorch**, you may want to cite the following [paper](https://uf
 
 ## Installation
 
-`nmtpytorch` currently requires `python>=3.6` and `torch==0.4.1`.
+`nmtpytorch` currently requires `python>=3.6` and `torch>=0.4.1`.
 We are not planning to support Python 2.x.
 
 **IMPORTANT:** After installing `nmtpytorch`, you **need** to run `nmtpy-install-extra`
@@ -58,7 +58,7 @@ $ pip install nmtpytorch
 ```
 
 This will automatically fetch and install the dependencies as well. For the `torch`
-dependency it will specifically install the `torch 0.4.1` package from `PyPI` that
+dependency it will specifically install the `torch 1.0.0` package from `PyPI` that
 ships `CUDA 9.0` within.
 
 ### conda

setup.py

@@ -45,7 +45,7 @@ def get_nmtpytorch_version():
     python_requires='~=3.6',
     install_requires=[
         'numpy', 'scipy', 'scikit-learn', 'tqdm', 'pillow',
-        'torch>=0.4.1', 'torchvision==0.2.1',
+        'torch==1.0.0', 'torchvision==0.2.1',
         'sacrebleu>=1.2.9', 'tensorboardX==1.1',
         'editdistance==0.4', 'subword_nmt==0.3.5',
     ],