pytorch cpu版和gpu版不能装在一个环境中

我之前环境里装了一个cpu版的pytorch,训练网络太慢了我就想再装一个gpu版的。

运行了这样的操作之后,用这个环境运行

import torch
print(torch.__version__) # torch version
print(torch.version.cuda)  # cuda version
print(torch.backends.cudnn.version())  # cudnn version

这串代码

得到结果

说明还是在运行cpu版的pytorch

因此想cpu换gpu的要么删除环境中的torch库(有一大堆),要么新建环境 


版权声明:本文为qq_56671474原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。