开发者社区 > ModelScope模型即服务 > 正文

ModelScope中torch包的版本要求多少?

ModelScope中torch包的版本要求多少?

展开
收起
小小爱吃香菜 2024-05-01 13:01:44 171 0
3 条回答
写回答
取消 提交回答
  • odelScope Library目前支持tensorflow,pytorch深度学习框架进行模型训练、推理, 在Python 3.8+, Pytorch 1.11+, Tensorflow上测试可运行。

    注: 大部分语音模型当前需要在Linux环境上使用,并且推荐使用python3.8 + tensorflow 2.13.0 + torch 2.0.1 的组合。部分模态模型可以在mac,windows等环境上安装使用,少部分模型需要tensorflow1.15.0。

    部分cv模型目前使用mmcv1.7.0,可以通过如下方式安装:

    目前只支持python3.10 torch支持2.1.0和2.1.1,cuda支持11.8.0,12.1.0
    对应版本1.7.0+torch2.1.1cu121 1.7.0+torch2.1.0cu121 1.7.0+torch2.1.1cu118 1.7.0+torch2.1.0cu118
    pip install mmcv_full=='1.7.0+torch2.1.1cu121' -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html

    image.png

    参考文档https://modelscope.cn/docs/%E7%8E%AF%E5%A2%83%E5%AE%89%E8%A3%85

    2024-05-02 09:03:23
    赞同 展开评论 打赏
  • 最新的镜像包torch版本是2.1.2。
    image.png
    大部分语音模型当前需要在Linux环境上使用,并且推荐使用python3.8 + tensorflow 2.13.0 + torch 2.0.1 的组合。
    image.png

    ——参考链接

    2024-05-02 08:13:21
    赞同 展开评论 打赏
  • 2.1.2 。 此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2024-05-01 16:13:37
    赞同 展开评论 打赏

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352

热门讨论

热门文章

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载