TypeError: argument 0 is not a Variable
问题定位:
- Variables and tensors:需要将输入数据转换为 Variable 类型。
解决方案:
- 升级 PyTorch 版本:In the current pytorch version (0.4.0) Variables and tensors are merged, so that shouldn’t be an error.(参考网址:https://ptorch.com/news/37.html)
## 查看 PyTorch 版本
>>> import torch
>>> print(torch.__version__)
0.3.1
## 更新 PyTorch 版本
##
pip install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp27-cp27mu-linux_x86_64.whl
pip install torchvision
# if the above command does not work, then you have python 2.7 UCS2, use this command
pip install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp27-cp27m-linux_x86_64.whl
按照上述步骤将 PyTorch 升级到 0.4.1 版本后,就解决了上述问题。
相关参考资料:
- PyTorch-generative-model-collections:https://github.com/znxlwm/pytorch-generative-model-collections
- https://discuss.pytorch.org/t/nn-tutorial-argument-0-is-not-a-variable/4606/11
- python优先的深度学习框架:https://ptorch.com/