gpt4 book ai didi

python - 正确地将 NumPy 数组转换为在 GPU 上运行的 PyTorch 张量

转载 作者:行者123 更新时间:2023-12-01 08:19:21 26 4
gpt4 key购买 nike

我创建了一个看起来像这样的DataLoader

class ToTensor(object):
def __call__(self, sample):
return torch.from_numpy(sample).to(device)

class MyDataset(Dataset):
def __init__(self, data, transform=None):
self.data = data
self.transform = transform

def __len__(self):
return len(self.data)

def __getitem__(self, idx):
sample = self.data[idx, :]

if self.transform:
sample = self.transform(sample)

return sample

我正在像这样使用这个数据加载器

dataset = MLBDataset(
data=data,
transform=transforms.Compose([
ToTensor()
]))
dataloader = DataLoader(dataset, batch_size=BATCH_SIZE, shuffle=True, num_workers=4)
dataiter = iter(dataloader)
x = dataiter.next()

失败并显示消息

THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1549628766161/work/aten/src/THC/THCGeneral.cpp line=55 error=3 : initialization error
THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1549628766161/work/aten/src/THC/THCGeneral.cpp line=55 error=3 : initialization error
THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1549628766161/work/aten/src/THC/THCGeneral.cpp line=55 error=3 : initialization error
THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1549628766161/work/aten/src/THC/THCGeneral.cpp line=55 error=3 : initialization error
...
torch._C._cuda_init()
RuntimeError: cuda runtime error (3) : initialization error at /opt/conda/conda-bld/pytorch_1549628766161/work/aten/src/THC/THCGeneral.cpp:55

对于 ToTensor() 中的 return 命令,事实上,任何在该类中移动 GPU 张量的尝试都会失败。我尝试过:

a = np.array([[[1, 2, 3, 4], [5, 6, 7, 8], [25, 26, 27, 28]],
[[11, 12, np.nan, 14], [15, 16, 17, 18], [35, 36, 37, 38]]])
print(torch.from_numpy(a).to(device))

ToTensor() 中的 __call__ 主体内,它失败并显示相同的消息,而在其他地方都成功。

为什么会产生此错误以及如何解决此错误?

最佳答案

试试这个:

代码:

import numpy as np
import torch
import torch.nn as nn

torch.cuda.set_device(0)

X = np.ones((1, 10), dtype=np.float32)
print(type(X), X)
X = torch.from_numpy(X).cuda(0)
print(type(X), X)

model = nn.Linear(10, 10).cuda(0)
Y = model(X)
print(type(Y), Y)

输出:

<class 'numpy.ndarray'> [[1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]]
<class 'torch.Tensor'> tensor([[1., 1., 1., 1., 1., 1., 1., 1., 1., 1.]], device='cuda:0')
<class 'torch.Tensor'> tensor([[ 0.4867, -1.0050, 0.4872, -0.0260, -0.0788, 0.0161, 1.2210, -0.3957,
0.2097, 0.2296]], device='cuda:0', grad_fn=<AddmmBackward>)

关于python - 正确地将 NumPy 数组转换为在 GPU 上运行的 PyTorch 张量,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54773293/

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