atp*_*atp 4 gpu deep-learning pytorch
我正在使用配备 Intel Corporation HD Graphics 520 的笔记本电脑。有谁知道如何为深度学习设置它,特别是 Pytorch?我已经看到如果你有 Nvidia 显卡我可以安装 cuda 但是当你有英特尔 GPU 时该怎么办?
PyTorch 不支持 NVIDIA CUDA 和最近的 AMD Rocm 以外的任何东西。英特尔在其他答案中对 Pytorch 的支持仅适用于至强系列处理器,并且在 GPU 方面也没有那么可扩展。然而,
英特尔oneAPI以前的广告oneDNN支持包括英特尔集成显卡在内的各种硬件,但目前,截至 2020 年 10 月 29 日,PyTorch 或 PyTorch 1.7 尚未实现全面支持。
但你还有其他选择。推断你有几个选择。
DirectML就是其中之一。基本上你把你的模型转换成onnx,然后使用 directml provider 在 gpu 上运行您的模型(在我们的例子中将使用 DirectX12 并且目前仅适用于 Windows!)您的另一个选择是使用OpenVino和TVM,它们都支持多平台,包括 Linux、Windows、Mac等
。它们都使用ONNX模型,因此您需要先将模型转换为onnx格式,然后再使用它们。
| 归档时间: |
|
| 查看次数: |
3730 次 |
| 最近记录: |