Fix setup, 0.2.1

environment.yml

@@ -2,5 +2,4 @@ name: gpytorch
 channels:
   - pytorch
 dependencies:
-  - pytorch-nightly
-  - torchvision
+  - pytorch

gpytorch/__init__.py

@@ -35,7 +35,7 @@
 from .lazy import lazify, delazify
 
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 
 # Old deprecated stuff
 fast_pred_var = beta_features._moved_beta_feature(settings.fast_pred_var, "gpytorch.settings.fast_pred_var")

setup.py

@@ -24,9 +24,8 @@ def find_version(*file_paths):
     raise RuntimeError("Unable to find version string.")
 
 
-readme = open('README.md').read()
-version = find_version('gpytorch', '__init__.py')
-
+readme = open("README.md").read()
+version = find_version("gpytorch", "__init__.py")
 
 # See if we have the development version of pytorch installed
 # We will skip installing the stable version of PyTOrch if this is true
@@ -37,13 +36,14 @@ def find_version(*file_paths):
     has_dev_pytorch = False
 
 # Base equirements
-with open("requirements.txt", "r") as f:
-    install_requires = f.read().strip().split("\n")
-    if has_dev_pytorch:  # Remove the PyTorch requirement
-        install_requires = [
-            install_require for install_require in install_requires
-            if "torch" != re.split(r"(=|<|>)", install_require)[0]
-        ]
+install_requires = [
+    "torch>=1.0.0",
+]
+if has_dev_pytorch:  # Remove the PyTorch requirement
+    install_requires = [
+        install_require for install_require in install_requires
+        if "torch" != re.split(r"(=|<|>)", install_require)[0]
+    ]
 
 # Run the setup
 setup(