Skip to content
This repository was archived by the owner on Apr 8, 2025. It is now read-only.

Update PyTorch to 1.5#351

Merged
tanaysoni merged 1 commit intomasterfrom
torch-15
May 6, 2020
Merged

Update PyTorch to 1.5#351
tanaysoni merged 1 commit intomasterfrom
torch-15

Conversation

@tanaysoni
Copy link
Copy Markdown
Contributor

No description provided.

@tanaysoni tanaysoni requested a review from tholor May 6, 2020 14:54
@tholor
Copy link
Copy Markdown
Member

tholor commented May 6, 2020

Did you verify that the GPUs can be found in our new docker based on nvidia/cuda:10.2-base-ubuntu18.04? I know that with the old base image it was not working due to cuda upgrade in pytorch 1.5.0

@tanaysoni
Copy link
Copy Markdown
Contributor Author

Yes, this image works on V100 GPU.

@tanaysoni tanaysoni merged commit ab9ad89 into master May 6, 2020
@tanaysoni tanaysoni deleted the torch-15 branch May 6, 2020 16:22
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.

2 participants