gpt4 book ai didi

PyTorch ROCm 已出 - 如何选择 Radeon GPU 作为设备

转载 作者:行者123 更新时间:2023-12-05 03:43:07 24 4
gpt4 key购买 nike

由于 Pytorch 发布了 ROCm 版本,这使我能够使用 nvidias 以外的其他 gpu,我如何在 python 中选择我的 radeon gpu 作为设备?显然,像 device = torch.cuda.is_available 或 device = torch.device("cuda") 这样的代码不起作用。谢谢你的帮助。 :)

最佳答案

我花了很长时间才弄明白。我的 AMD GPU 仍然存在一些配置问题,所以我无法测试它是否有效,但是,根据这个 github pytorch 线程,Rocm 集成是这样编写的,所以你可以只需调用 torch.device('cuda') 即可,无需实际移植!

在这里查看主题:https://github.com/pytorch/pytorch/issues/10670

关于PyTorch ROCm 已出 - 如何选择 Radeon GPU 作为设备,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/66909581/

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