please use torch.load with map_location=torch.device('cpu')

后端 未结 0 1876
爱一瞬间的悲伤
爱一瞬间的悲伤 2021-01-25 17:39

I am running Python program, but I do not have a GPU, what can I do to make Python use CPU instead of GPU?

$ python extract_feature.py --data mnist --net checkpoint_4.pth

相关标签:
回答
  • 消灭零回复
提交回复
热议问题