如何将AMD GPU用于fastai/pytorch?

如何将AMD GPU用于fastai/pytorch?,pytorch,gpu,amd,fast-ai,Pytorch,Gpu,Amd,Fast Ai,我正在使用一台笔记本电脑,它有Intel Corporation HD Graphics 5500(rev 09)和AMD Radeon r5 m255图形卡 是否有人知道如何为深度学习设置it,特别是fastai/Pytorch?更新: 自Pytorch 1.8版(2021年3月4日)起,AMD ROCm版本可从Pytorch官方网站获得。您现在可以轻松地在Linux和Mac上安装它们,与安装CUDA/CPU版本的方式相同 目前,仅提供pip包。此外,Mac和Windows平台仍然不受支持(不

我正在使用一台笔记本电脑,它有Intel Corporation HD Graphics 5500(rev 09)和AMD Radeon r5 m255图形卡

是否有人知道如何为深度学习设置it,特别是fastai/Pytorch?

更新: 自Pytorch 1.8版(2021年3月4日)起,AMD ROCm版本可从Pytorch官方网站获得。您现在可以轻松地在Linux和Mac上安装它们,与安装CUDA/CPU版本的方式相同

目前,仅提供pip包。此外,Mac和Windows平台仍然不受支持(不过我还没有使用WSL2进行测试!)

旧答案: 您需要安装ROCm版本。AMD关于建造Pytorch的官方指示如下

以前有一个rocm的轮子包,但是AMD似乎不再发布了,相反,你需要从源代码中构建PyTorch,正如我链接到的指南所解释的那样


但是,您可以查阅此页面,以构建最新的PyTorch版本:。

不了解PyTorch,但是,即使Keras现在与TF集成,您也可以使用库PlaidML在AMD GPU上使用Keras!由英特尔公司制造。它非常酷,易于设置,而且可以方便地为不同的项目切换Keras后端

是否支持Vega 11?我只在名单上看到了织女星10号。我不知道,他们的回购协议似乎有问题,可能值得一看