解决RuntimeError: Can‘t call numpy() on Variable that requires grad. Use var.detach().numpy() instead.

转自：https://blog.csdn.net/discoverer100/article/details/103902130

1. 问题描述

如题，将PyTorch Tensor类型的变量转换成numpy时报错：

RuntimeError: Can't call numpy() on Variable that requires grad. Use var.detach().numpy() instead.

•  

2. 解决办法

出现这个现象的原因是：待转换类型的PyTorch Tensor变量带有梯度，直接将其转换为numpy数据将破坏计算图，因此numpy拒绝进行数据转换，实际上这是对开发者的一种提醒。如果自己在转换数据时不需要保留梯度信息，可以在变量转换之前添加detach()调用。假设原来的写法是：

aaa.cpu().numpy()

•  

那么现在改为

aaa.cpu().detach().numpy()

•  

即可。

 

---

 

参考资料：

• https://discuss.pytorch.org/t/cant-call-numpy-on-variable-that-requires-grad-use-var-detach-numpy-instead/28281/3
• https://discuss.pytorch.org/t/cant-call-numpy-on-variable-that-requires-grad/20763