numpyのにCUDAテンソルを変換することはできません。使用Tensor.cpu()最初のホストメモリへのテンソルをコピーします

分析:CUDAでのデータ()、あなただけのCPU上で、numpyのに型テンソルをフロートする必要があります

方法を変更する前に:

labels_=labels_.numpy().tolist()

 ああ、変更後に見て

labels_=labels_.cpu().numpy().tolist()

 

公開された234元の記事 ウォン称賛61 ビュー120 000 +

おすすめ

転載: blog.csdn.net/weixin_42528089/article/details/104926787