安装适用于 Windows 11 和 AMD GPU 的 Pytorch

问题描述 投票:0回答:1

有人可以帮我安装 Pytorch 吗?我的设备当前使用 Windows 操作系统和 AMD GPU。但是,Pytorch 安装不支持与 ROCm 组合的 Windows 操作系统。只有选择 Linux 操作系统时,ROCm 选项才可用。

我可以使用 CUDA 工具包来替代 ROCm 吗?或者我是否可以将我的操作系统更改为Linux?有没有办法绕过所有这些并且仍然能够使用 Pytorch?

任何建议将不胜感激!

我尝试在 youtube 上寻找安装教程,但他们没有与我相同的操作系统和 GPU 组合。 (即Windows操作系统和AMD GPU)

windows machine-learning pytorch amd
1个回答
0
投票

切换到 Linux 当然可行,但似乎没有必要。 如果您选择这条路线,我建议您从Ubuntu开始,因为有很多可用的信息。

CUDA 工具包仅在您拥有 NVIDIA GPU 时可用,并且不适合您。

要让 Pytorch 在 Windows 上工作,请查看这个堆栈溢出问题,因为它非常详细:How to use AMD GPU for fastai/pytorch?

© www.soinside.com 2019 - 2024. All rights reserved.