我跑步时
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
,但仍然不起作用。
任何帮助将不胜感激!