這幾天在使用vot toolkit評測自己的跟蹤器,需要在matlab裏面調用python的跟蹤器代碼,其中用到了pytorch的庫。運行的時候報錯,“import error: from torch._C import * RuntimeError: stoi”,搞了好久才解決。查了一些資料,顯示這個問題是pytorch版本導致的,如果使用Torch 0.3.*則沒有這個問題。但是將pytorch降級肯定是行不通的,畢竟很多庫函數都不一樣了,改起來更麻煩。
這個問題可以通過兩種方式來解決,經過嘗試二者選其一即可,方案如下:
1. 在調用python的指令前,加上“env -i”,示例如下:
system('env -i /usr/bin/python3 -c "import torch"')
2. 在python的腳本文件中加上如下指令:
del os.environ['MKL_NUM_THREADS']
import torch
這個問題的解決主要參考了github上pytorch上面問題的討論,鏈接如下:
https://github.com/pytorch/pytorch/issues/7082
希望能對碰到相同問題的人提供幫助,如果有類似問題也可以互相交流。