为什么在pytorch中出现RuntimeError：CUDA错误：无效参数？

Question

最近，我在调用类似RuntimeError: CUDA error: invalid argument之类的函数时经常得到torch.cholesky，例如：

import torch
a = torch.randn(3, 3, device="cuda:0")
a = torch.mm(a, a.t()) # make symmetric positive-definite
torch.cholesky(a)

如果我改用device="cpu"，这可以正常工作。该错误的描述性不是很高，所以我不确定这里出了什么问题。

Answer 1

我发现此错误是因为运行我的计算机上已安装了CUDA 10，但我只是将pytorch安装为pip install torch。在他们的website中，pip是安装pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-linux_x86_64.whl和CUDA 10的正确方法。