pytorch中常見錯誤總結

2021-10-04 15:55:03 字數 653 閱讀 7134

錯誤1:在console中輸入import torchvision會報錯:importerror: cannot import name 'pillow_version'

原因:torchvision和pillow版本不相容,可能pillow版本過高,如pillow 7.0和torchvision 0.3.0衝突。

解決辦法:降低pillow版本,即輸入:

pip install pillow==6.2.2
錯誤2:indexerror: invalid index of a 0-dim tensor. use tensor.item() to convert a 0-dim tensor to a python

原因:train_loss += loss.data[0] 是pytorch0.3.1版本**,在0.4-0.5版本的pytorch會出現警告,不會報錯,但是0.5版本以上的pytorch就會報錯,總的來說是版本更新問題.

解決辦法:

loss.data[0]修改為loss.item()

python中常見錯誤

1.手誤造成的名字錯誤 nameerror 名字錯誤 如 nameerror name pint is not defined 2.多行 不能一行寫 syntaxerror 語法錯誤 如 syntaxerror invalid syntax 3.縮排錯誤 indentationerror unexp...

pytorch安裝 及 常見錯誤總結(長期更新)

分割線 pip install torch.whl conda install torchvision 0.2.1 cudatoolkit 8.0其中第二部conda install 可以事先選擇清華源 sudo rm rf nv 有乙個變數不是tensor型別 很有可能當前安裝scipy版本為1....

基於C 中常見記憶體錯誤的總結

在系統開發過程中出現的bug相對而言是比較好解決的,花費在這個上面的除錯代價不是很大,但是在系統整合後的bug往往是難以定位的bug 最好方式是打樁,通過打樁可以初步鎖定出錯的位置,如 進入函式前列印日誌,離開時再次列印日誌 而這些難以定位的bug基本分為2類 記憶體錯誤和並非問題。1 記憶體洩露 ...