torch.from_numpy()报错:TypeError: expected np.ndarray (got numpy.ndarray)

看到有人说把torch.from_numpy()改成torch.Tensor(),我试了下确实可以,但是仅限于你只有这一个地方报错,如果用到torchvision.transforms之类的库,只要里面有从numpy转torch的操作就会报错
后来发现是因为numpy版本太高,我的是2.0.0,改成1.16.4之后就好了

posted @ 2024-07-05 10:11  钢之炼丹术师  阅读(1446)  评论(0)    收藏  举报
努力加载评论中...
点击右上角即可分享
微信分享提示