gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-04 04:20:28 31 4
gpt4 key购买 nike

我在尝试使用 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 运行得很好。

最佳答案

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

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

关于tensorflow - 不能将 GPU 与 Pytorch 一起使用,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59506935/

31 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com