tensorflow - 不能将 GPU 与 Pytorch 一起使用

Question

我在尝试使用 Pytorch 时不断收到此错误。

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

我使用 conda install pytorch torchvision cudatoolkit=10.1 -c pytorch 安装了 Pytorch。

使用 tensorflow 我的 GPU 运行得很好。

Answer 1

您可以通过安装 CUDA 10.2(最新版本)修复此错误，另外使用此命令重新安装 Pytorch:

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch