Pytorch Cuda运行时错误(正确安装了驱动程序)

时间:2019-09-20 06:25:20

标签: gpu pytorch nvidia

我跑步时

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(拉伸)。在控制台上检查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,但仍然不起作用。

任何帮助将不胜感激!

0 个答案:

没有答案