不能将GPU与火炬一起使用
不能将GPU与火炬一起使用
提问于 2019-12-27 23:24:16
我一直在尝试使用Py手电筒时出现这个错误。
RuntimeError:试图在CUDA设备上反序列化对象，但是torch.cuda.is_available()是假的。如果您在一台只运行CPU的机器上运行，请使用torch.load和map_location=torch.device(“CPU”)将您的存储映射到CPU。
我使用conda install pytorch torchvision cudatoolkit=10.1 -c pytorch安装了Pytorch。
有了tensorflow，我的GPU运行得很好。
回答 2
Stack Overflow用户
发布于 2020-04-28 20:54:45
您可以通过安装CUDA 10.2 (最新版本)来修复此错误，另外，还可以使用以下命令重新安装Pytorch：
conda install pytorch torchvision cudatoolkit=10.2 -c pytorchStack Overflow用户
发布于 2020-05-01 17:23:42
使用来自pytorch.org的正式安装命令
对于Windows+GPU (对于其他人，请查看pytorch.org)
pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/59506935
复制相关文章
相似问题
腾讯云开发者
