小编elh*_*ili的帖子

请使用 torch.load 和 map_location=torch.device('cpu')

我正在运行Python程序,但我没有GPU,我该怎么做才能让Python使用CPU而不是GPU?

$ python extract_feature.py --data mnist --net checkpoint_4.pth.tar --features pretrained
Run Code Online (Sandbox Code Playgroud)

它给了我以下警告:

=> 运行时错误:尝试在 CUDA 设备上反序列化对象,但 torch.cuda.is_available() 为 False。如果您在仅使用 CPU 的计算机上运行,​​请使用 torch.load 和 map_location=torch.device('cpu') 将存储映射到 CPU。

照片是我的Python项目的结构:

1

python data-mining deep-learning

5
推荐指数
1
解决办法
3万
查看次数

标签 统计

data-mining ×1

deep-learning ×1

python ×1