1. 程式人生 > >【PyTorch】TypeError: argument 0 is not a Variable

【PyTorch】TypeError: argument 0 is not a Variable

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 版本後,就解決了上述問題。

 

相關參考資料: