Pytorch Cuda Runtime Ошибка (с правильно установленными драйверами) - PullRequest
Новая
       96

Pytorch Cuda Runtime Ошибка (с правильно установленными драйверами)

2 голосов
/ 20 сентября 2019

Когда я запускаю 

import torch
torch.cuda.FloatTensor([1.])

Я получаю сообщение об ошибке: RuntimeError: cuda runtime error (38) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:50.

Я использую Debian (Stretch).Проверка драйвера nvidia на консоли: 

/sbin/modinfo nvidia_current

возвращает: 

filename:       /lib/modules/4.9.0-11-amd64/updates/dkms/nvidia-current.ko
alias:          char-major-195-*
version:        418.74
supported:      external
license:        NVIDIA
srcversion:     AB4044DE27C9CA55579A110

, когда я nvcc --version, я получаю: 

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Tue_Jun_12_23:07:04_CDT_2018
Cuda compilation tools, release 9.2, V9.2.148

На Python, когдаЯ делаю 

import torch
print(torch.version.cuda)  
print(torch.cuda.device_count())
print(torch.cuda.is_available())

Я получаю: 

9.2.148
0
False

Я пытался сделать os.environ["CUDA_VISIBLE_DEVICES"] = '0, но это все равно не работает.

Любая помощь будет принята с благодарностью!

...