最近更新了pytorch,直接用的
pip install --upgrade torch torchvision
发现运行原来的代码报错了,不能设置cuda(),第一反应就是更新导致cuda版本和torch版本不对应,查了下果然如此,实验室服务器cuda版本是0.8.16,所以直接卸载了torch重新装,从pytorch官网复制指令非常稳健。
所以以后更新还是乖乖的在官网去复制指令吧~
最近更新了pytorch,直接用的
pip install --upgrade torch torchvision
发现运行原来的代码报错了,不能设置cuda(),第一反应就是更新导致cuda版本和torch版本不对应,查了下果然如此,实验室服务器cuda版本是0.8.16,所以直接卸载了torch重新装,从pytorch官网复制指令非常稳健。
所以以后更新还是乖乖的在官网去复制指令吧~