Skip to content

Conversation

@zou3519
Copy link
Contributor

@zou3519 zou3519 commented Dec 14, 2017

Running on some V100 gpus, I saw the following message:

  warnings.warn(error_str % (d, name, 8000, CUDA_VERSION))
/private/home/rzou/pytorch/pytorch/torch/cuda/__init__.py:89: UserWarning:
    Found GPU7 Tesla V100-SXM2-16GB which requires CUDA_VERSION >= 8000 for
     optimal performance and fast startup time, but your PyTorch was compiled
     with CUDA_VERSION 8000. Please install the correct PyTorch binary
     using instructions from http://pytorch.org

This fixes it so that it'll say "which requires CUDA_VERSION >= 9000"

@soumith soumith merged commit d450895 into pytorch:master Dec 14, 2017
@zou3519 zou3519 deleted the typo branch January 3, 2018 19:50
@soumith soumith added the 0.3.1 label Feb 4, 2018
soumith pushed a commit that referenced this pull request Feb 7, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants