首先检查pytorch、python、cuda版本之间的对应关系
重点关注cuda的版本!!!
其次,如果cuda版本正确,检查python版本,这是最隐蔽且最容易是问题所在的地方。
可以尝试更改Python版本
例:
我的配置:
torch 1.0.1
cuda 10.0
python 3.6.15
一直报错,最后将Python版本改为3.6.2,程序正常运行。
首先检查pytorch、python、cuda版本之间的对应关系
重点关注cuda的版本!!!
其次,如果cuda版本正确,检查python版本,这是最隐蔽且最容易是问题所在的地方。
可以尝试更改Python版本
例:
我的配置:
torch 1.0.1
cuda 10.0
python 3.6.15
一直报错,最后将Python版本改为3.6.2,程序正常运行。