Почему я получаю RuntimeError: ошибка CUDA: неверный аргумент в pytorch?

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. С их веб-сайта правильный способ установки с pip и CUDA 10 - pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-linux_x86_64.whl.