在Colaboratory中,无法使用CUDA进行torch操作。

17

错误信息如下:

RuntimeError   Traceback (most recent call last)
<ipython-input-24-06e96beb03a5> in <module>()
     11
     12 x_test = np.array(test_features)
---> 13 x_test_cuda = torch.tensor(x_test, dtype=torch.float).cuda()
     14 test = torch.utils.data.TensorDataset(x_test_cuda)
     15 test_loader = torch.utils.data.DataLoader(test, batch_size=batch_size, shuffle=False)

/usr/local/lib/python3.6/dist-packages/torch/cuda/__init__.py in _lazy_init()
    160 class CudaError(RuntimeError):
    161     def __init__(self, code):
--> 162         msg = cudart().cudaGetErrorString(code).decode('utf-8')
    163         super(CudaError, self).__init__('{0} ({1})'.format(msg, code))
    164

RuntimeError: cuda runtime error (38) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:51

你能澄清一下你到目前为止尝试了什么吗?确切的设置是什么,你的相关代码是什么? - dennlinger
你检查过是否启用了GPU加速吗? - MBT
是的,我不打开这个GPU!谢谢! - biao_biao
1
你所说的“the torch”,是指PyTorch吗?“Torch”标签的维基百科页面上写道:“Torch是一个LuaJIT科学计算框架,被全球机器学习研究人员广泛使用。”。 - Peter Mortensen
2个回答

62

点击运行时,选择更改运行时类型

现在在硬件加速中,选择GPU并点击保存


实际拼写是“更改运行时类型”吗? - Peter Mortensen
@PeterMortensen 是的 - lucidbrot

-1

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接