Skip to content

Replaced pytorch cuda by cpu installation #2640

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 1 commit into from
Sep 2, 2020
Merged

Replaced pytorch cuda by cpu installation #2640

merged 1 commit into from
Sep 2, 2020

Conversation

@vfdev-5
Copy link
Contributor

@vfdev-5 vfdev-5 commented Sep 2, 2020 •
edited
Loading

Description:

  • On Travis CI there is no GPU and thus we can download pytorch cpu only version for tests instead of one with cuda.

Installation times (conda create -q -n test-environment pytorch ... -c pytorch-nightly):

  • before : 170s
  • now : 21s

fmassa
fmassa approved these changes Sep 2, 2020
View reviewed changes
Copy link
Member

@fmassa fmassa left a comment

Choose a reason for hiding this comment

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

Awesome, thanks!

@fmassa fmassa merged commit 02f46a5 into pytorch:master Sep 2, 2020
@vfdev-5 vfdev-5 deleted the travis-ci-pytorch-cpu-only branch September 2, 2020 10:40
bryant1410 pushed a commit to bryant1410/vision-1 that referenced this pull request Nov 22, 2020
@vfdev-5 @bryant1410
Replaced pytorch cuda by cpu installation (pytorch#2640)
9b22f59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@fmassa fmassa fmassa approved these changes

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@vfdev-5 @fmassa