复制
收藏
提问
简洁

pytorch不再支持rocm了吗

5ddb9a101
0秒前
全网
文档
学术
百科
知识库
时间不限
自动
回答简洁模式
深入研究一些

PyTorch 继续支持 ROCm 平台。

支持情况

  • ROCm 支持起始:PyTorch 1.8 版本开始支持 ROCm,允许用户在原生 Linux 环境下运行 PyTorch,无需 Docker 配置。4
  • ROCm 支持升级:到了 PyTorch 1.12 版本,ROCm 的支持从测试版升级为稳定版,进一步增强了在原生环境中运行 PyTorch 的便利性。2
  • 安装选项:用户可以通过访问 pytorch.org/get-started/locally/ 并选择 "ROCm" 计算平台来获取测试过的 wheel 包进行安装。357

兼容性与资源

  • 兼容性矩阵:PyTorch 官网提供了一个兼容性矩阵,展示了 ROCm 与 PyTorch 构建之间的安装兼容性。5
  • 资源访问:研究人员和开发人员可以在 Ubuntu® Linux® 上使用 AMD ROCm 5.7,利用基于 AMD RDNA™ 3 GPU 架构的优势。8
你觉得结果怎么样?
深度学习框架有哪些?
ROCm平台支持哪些应用?
如何优化GPU在深度学习中的性能?
PyTorch 1.8 版本支持 ROCm 吗?
ROCm 支持升级到稳定版了吗?
如何在 Ubuntu Linux 上安装 PyTorch ROCm 版本?

以上内容由AI搜集生成,仅供参考

在线客服