gpt4 book ai didi

numpy - TypeError : can’t convert CUDA tensor to numpy. 首先使用 Tensor.cpu() 将张量复制到主机内存

转载 作者:行者123 更新时间:2023-12-04 22:39:32 26 4
gpt4 key购买 nike

我正在使用 modified predict.py用于测试 pruned SqueezeNet Model

[phung@archlinux SqueezeNet-Pruning]$ python predict.py --image 3_100.jpg --model model_prunned --num_class 2
prediction in progress
Traceback (most recent call last):
File “predict.py”, line 66, in
prediction = predict_image(imagepath)
File “predict.py”, line 52, in predict_image
index = output.data.numpy().argmax()
TypeError: can’t convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
[phung@archlinux SqueezeNet-Pruning]$
我知道 numpy 还不支持 GPU。
我应该如何修改代码以在不调用张量复制数据操作的情况下摆脱此错误, tensor.cpu() ?

最佳答案

改变
index = output.data.numpy().argmax()

index = output.cpu().data.numpy().argmax()
这意味着数据首先移动到 cpu,然后转换为 numpy 数组

关于numpy - TypeError : can’t convert CUDA tensor to numpy. 首先使用 Tensor.cpu() 将张量复制到主机内存,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53900910/

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