【PyTorch】TypeError: argument 0 is not a Variable

TypeError: argument 0 is not a Variable

【PyTorch】TypeError: argument 0 is not a Variable_第1张图片

问题定位:

  • 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】TypeError: argument 0 is not a Variable_第2张图片

## 查看 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/

你可能感兴趣的:(【深度学习】,【python】)