开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

ModelScope中,我就是这样挂载的gpu,服务器的nvidia驱动和容器的驱动需要一致么?

ModelScope中,使用社区的cuda docker,在服务器中挂载了gpu,nvidia-smi可以看到当前gpu,但是torch还是使用不了cuda,9a4a3ae5c13e26861acf419f49538ba8.png
我就是这样挂载的gpu,服务器的nvidia驱动和容器的驱动需要一致么

展开
收起
多麻辣哦 2023-11-08 20:59:03 75 0
1 条回答
写回答
取消 提交回答
  • 您参考一下这个docker run --gpus all -p 8000:8000 -it -d --name ms registry.cn-hangzhou.aliyuncs.com/modelscope-repo/modelscope:ubuntu20.04-cuda11.8.0-py38-torch2.0.1-tf2.13.0-1.9.3 /bin/bash——此回答整理自钉群:魔搭ModelScope开发者联盟群 ①

    2023-11-08 22:00:51
    赞同 展开评论 打赏

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
冬季实战营第四期:零基础容器技术实战 立即下载