如果显示cuda和pytorch版本不对应
1.查一下torch.cuda.is_available()如果是False
2.conda list (more)查看下载的pytorch是cpu版的还是gpu版的。
3.可以考虑重新建个虚拟环境重新来
如果显示cuda和pytorch版本不对应
1.查一下torch.cuda.is_available()如果是False
2.conda list (more)查看下载的pytorch是cpu版的还是gpu版的。
3.可以考虑重新建个虚拟环境重新来