请问您是参考帮助文档的这块内容吗？
先配置python环境，然后再配置显卡驱动，然后再配置CUDA及其加速库；其中python环境配置您是选择单独下载的GPU环境的python组件包吗？
因为iDesktopX中的扩展包默认是包含关于机器学习Python的CPU运行环境，所以您在训练模型时使用的还是CPU。
您需要参考这块内容重新下载python组件包，然后再配置您的python环境验证
