Skip to content
This repository has been archived by the owner on Dec 16, 2022. It is now read-only.

tweak torch requirement #4166

Merged
merged 5 commits into from
May 1, 2020
Merged

tweak torch requirement #4166

merged 5 commits into from
May 1, 2020

Conversation

epwalsh
Copy link
Member

@epwalsh epwalsh commented Apr 29, 2020

As it is the setup.py requires torch>1.3.1,<=1.5.0. But this excludes version 1.5.0+cu101 which is needed if you have cuda 10.1 and not cuda 10.2. So - as I found out today - if you have 1.5.0+cu101 installed and you try running an allennlp command it will completely fail with this:

image

@epwalsh epwalsh requested a review from dirkgr April 29, 2020 22:00
setup.py Outdated
@@ -50,7 +50,7 @@
license="Apache",
packages=find_packages(exclude=["*.tests", "*.tests.*", "tests.*", "tests"]),
install_requires=[
"torch>1.3.1,<=1.5.0",
"torch>1.3.1,<1.5.1",
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

What about <1.6.0? PyTorch "patch" version changes shouldn't be breaking. But maybe you think it's better to have them CI-tested anyway?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Our CI will always test against the newest allowed package, so we'd know right away if a new patch version breaks anything. So yea, I agree with you, we should make it <1.6.0

@epwalsh epwalsh merged commit 31616de into allenai:master May 1, 2020
@epwalsh epwalsh deleted the torch-version branch May 1, 2020 18:37
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants