radeon rx 550x pytorch
时间: 2023-11-04 17:00:37 浏览: 145
pytorch
根据提供的引用内容,PyTorch 1.8正式支持AMD GPU,这意味着您可以在拥有AMD显卡的电脑上运行PyTorch深度学习框架。您可以通过以下命令在您的电脑上安装ROCm的PyTorch 1.8版本:
```
pip install torch -f https://download.pytorch.org/whl/rocm4.0.1/torch_stable.html
```
请注意,ROCm目前仅支持Ubuntu 20和18操作系统。如果您的系统版本较低,建议先升级到Ubuntu 20。确保您的显卡是AMD 580或更高版本。您可以通过检查内核来验证您的显卡是否支持ROCm。
阅读全文