PyTorch v1.3 с графическим процессором на Google Colab? - PullRequest
Новая
       52

PyTorch v1.3 с графическим процессором на Google Colab?

0 голосов
/ 20 октября 2019

Ноутбук Google colab поставляется с PyTorch и поддержкой графических процессоров - см. этот связанный ответ SO , а затем вам нужно изменить время работы ноутбука на GPU с CPU (из верхней строки меню Colab).

И все же попытка обучить модель из GPyTorch всегда терпит неудачу на 

/usr/local/lib/python3.6/dist-packages/gpytorch/utils/cholesky.py in psd_safe_cholesky(A, upper, out, jitter)
     19     """
     20     try:
---> 21         L = torch.cholesky(A, upper=upper, out=out)
     22         # TODO: Remove once fixed in pytorch (#16780)
     23         if A.dim() > 2 and A.is_cuda:

RuntimeError: CUDA error: invalid device function

Почему?

Соответствующая информация: 

import torch
torch.cuda.is_available()

True

print(torch.__version__)
print(gpytorch.__version__)
print(torch.backends.cudnn.version())

1.3.0+cu100
0.3.6
7603
...