指定神经网络训练应该使用带有 TargetDevice 选项的 GPU:
- GPU 训练仅适用于配备了 NVIDIA 显卡的 Windows 和 Linux 平台. 在这些平台上,TargetDevice"GPU" 实际上等价于 TargetDevice"CUDA".
- 为了充分利用 GPU 进行训练,最好为显卡配备最新的驱动程序.
- 如果指定 TargetDevice"GPU",而无兼容 GPU 可用,或显卡驱动程序未更新时,NetTrain 将给出出错消息.
指定神经网络训练应该使用带有 TargetDevice 选项的 GPU: