众所周知 PyTorch 已经可以通过 ROCm 支持AMD 的 GPU,奈何 ROCm 不支持 Windows。那么有没有办法在 Windows 上使用 AMD 的 GPU 来加速 PyTorch 呢?答案是肯定的,多谢 DirectML 和 Torch-DirectML。安装步骤请参考 Enable PyTorch with DirectML on Windows | Microsoft Learn。值得一提的是官方文档提及仅支持 PyTor...
使用miniforge安装的python。 安装CPU版本的pytorch: conda create-n torch conda activate torch conda install pytorch torchvision torchaudio cpuonly-c pytorch 使用GPU的话,需要安装directml pip install torch-directml 测试程序: importtorchimporttorch_directml dml=torch_directml.device()tensor1=torch.tensor([1...