gpt4 book ai didi

pytorch - 在 Colaboratory 中, torch 不能使用 CUDA

转载 作者:行者123 更新时间:2023-12-02 11:23:12 24 4
gpt4 key购买 nike

错误信息如下:

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

最佳答案

点击 运行时 并选择 更改运行时类型 .
现在在 硬件加速 ,选择 GPU 并点击保存。

关于pytorch - 在 Colaboratory 中, torch 不能使用 CUDA,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55368921/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com