能否在Foundry中使用PyTorch训练神经网络？外部训练后集成是否可行？

• 现在Foundry已经支持直接在平台内运行PyTorch训练任务，不用非得先外部训练再集成。具体可以这么操作：
  • 配置Python环境：在Code Workbooks或者Code Repositories中，通过requirements.txt添加PyTorch依赖，平台会自动搭建对应的运行环境。示例requirements.txt内容：

    torch>=2.0.0
    torchvision>=0.15.0
    

  • 调用平台计算资源：提交训练任务时，若平台支持GPU，可指定GPU节点，直接用Foundry的分布式集群运行训练脚本，训练数据能直接对接平台内的数据集，无需额外导出导入。
• 之前得到“无法实现”的结论，大概率是当时平台版本较旧或未开启相关配置。建议先查看当前Foundry的版本，或联系平台管理员确认PyTorch环境是否开放。
• 当然，如果你的训练场景有特殊需求（比如自定义CUDA版本、私有依赖库），外部训练后集成也是可行路径：将训练好的PyTorch模型导出为.pth或TorchScript格式，再在Foundry中通过Python代码加载模型，对接平台数据完成推理或后续处理。

内容的提问来源于stack exchange，提问作者G-Unit