我正在使用 PyTorch。我有以下代碼：import numpy as npimport torchX = np.array([[1, 3, 2, 3], [2, 3, 5, 6], [1, 2, 3, 4]])X = torch.DoubleTensor(X).cuda()X_split = np.array_split(X.numpy(), indices_or_sections = 2, axis = 0)X_split但我收到此錯誤：---------------------------------------------------------------------------TypeError Traceback (most recent call last)<ipython-input-121-870b5d3f67b6> in <module>()----> 1 X_prime_class_split = np.array_split(X_prime_class.numpy(), 2 indices_or_sections = 2, 3 axis = 0) 4 X_prime_class_splitTypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.錯誤消息很清楚，我知道如何通過僅包括.cpu(), ie 來修復此錯誤。X_prime_class.cpu().numpy(). 我只是想知道這是否證實了 numpy 數組不能在 GPU/Cuda 中運行？
1 回答
至尊寶的傳說
TA貢獻1789條經驗 獲得超10個贊
不，您通常不能在 GPU 陣列上運行 numpy 函數。PyTorch 為 PyTorch 張量重新實現了 numpy 中的大部分功能。例如，torch.chunk工作方式類似于np.array_split您可以執行以下操作：
我正在使用 PyTorch。我有以下代碼：
import numpy as np
import torch
X = np.array([[1, 3, 2, 3], [2, 3, 5, 6], [1, 2, 3, 4]])
X = torch.DoubleTensor(X).cuda()
X_split = np.array_split(X.numpy(),
indices_or_sections = 2,
axis = 0)
X_split
但我收到此錯誤：
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-121-870b5d3f67b6> in <module>()
----> 1 X_prime_class_split = np.array_split(X_prime_class.numpy(),
2 indices_or_sections = 2,
3 axis = 0)
4 X_prime_class_split
TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
錯誤消息很清楚，我知道如何通過僅包括.cpu(), ie 來修復此錯誤。X_prime_class.cpu().numpy(). 我只是想知道這是否證實了 numpy 數組不能在 GPU/Cuda 中運行？
添加回答
舉報
0/150
提交
取消