在非GPU机器上使用pickle加载.txt文件

Question

我正在尝试在非 GPU机器上加载.txt文件。 .txt文件已转储到GPU机器上：

with open("Data/all_rewards.txt", "wb") as f:
    pickle.dump(all_rewards, f)

该文件仅包含文本数据，没有网络模型或其他任何内容。 在GPU机器上，我可以轻松加载数据：

with open("all_rewards.txt", 'rb') as f:
    losses = pickle.load(f)

但是在非 GPU机器上，我收到错误消息：

AssertionError('Torch not compiled with CUDA enabled',)

由于Torch也使用pickle加载模型，因此我尝试使用：p加载数据

losses = torch.load("all_rewards.txt", map_location='cpu')

但是那没用。 你能帮我pla吗？

编辑： 这个问题以某种方式神奇地解决了自己。我已经重新安装了pickle，最初并不能解决问题，但是现在可以了...

Answer 1

听起来像您在非GPU机器上安装pytorch的方式存在问题。除了pip install torchvision

外，还需要运行命令

如果您使用操作系统配置前往https://pytorch.org/并选择CUDA：None，则应该会收到一个附加的http链接，以下载一个附加的车轮套件。

例如：Linux，pip和Python3.6产生以下附加命令

pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl