diff options
| author | Soumith Chintala <soumith@gmail.com> | 2019-02-06 15:46:39 -0800 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-02-06 15:51:58 -0800 |
| commit | 717ae09184a5329f53d99bc56d0ab0118c3522e6 (patch) | |
| tree | 3afc3dac6699d6076f7a505ea58044a239e7c1f6 | |
| parent | 8105aaca8611acd9e33707d6b57c6b1f144e4ab4 (diff) | |
| download | pytorch-717ae09184a5329f53d99bc56d0ab0118c3522e6.tar.gz pytorch-717ae09184a5329f53d99bc56d0ab0118c3522e6.tar.bz2 pytorch-717ae09184a5329f53d99bc56d0ab0118c3522e6.zip | |
improve error message (#16719)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/16712
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16719
Differential Revision: D13978688
Pulled By: ezyang
fbshipit-source-id: 61f8fa4c54c6969a58550e32e18be2eb9254ced7
| -rw-r--r-- | torch/cuda/__init__.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/torch/cuda/__init__.py b/torch/cuda/__init__.py index ef55f017fe..75e7272d7c 100644 --- a/torch/cuda/__init__.py +++ b/torch/cuda/__init__.py @@ -102,6 +102,7 @@ def _check_capability(): old_gpu_warn = """ Found GPU%d %s which is of cuda capability %d.%d. PyTorch no longer supports this GPU because it is too old. + The minimum cuda capability that we support is 3.5. """ CUDA_VERSION = torch._C._cuda_getCompiledVersion() |