pytorch版本的问题,架构本身的版本与实际的cuda环境以及想要跑通的代码不兼容;
我遇到的问题是将pytorch版本降低,降到torch.1.0版本,对应的cuda环境也比较低,需要去网上对应着查一下cuda版本以及对应的pytorch,和你所搭建的服务器Linux环境中的cuda版本是否存在兼容性问题。如果不存在兼容性问题,直接在服务器中完成环境的搭建过程即可。
pytorch.orghttps://pytorch.org/get-started/previous-versions/上面是个什么呢,是你需要查找的pytorch以及cuda对应的版本链接,需要从中进行查找自己的版本是否存在不兼容的情况。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)