我之前环境里装了一个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版权协议,转载请附上原文出处链接和本声明。