"在魔搭ModelScope社区下载了GPU镜像,并运行了容器。但是用nvidia-smi时提示。Can't initialize NVML
warnings.warn(""Can't initialize NVML"") ModelScope是不是在容器中还要安装nvidia的驱动?
"
参考这个命令,参考以下代码 docker run --gpus all -p 8000:8000 -it -d --name ms registry.cn-beijing.aliyuncs.com/modelscope-repo/modelscope:ubuntu22.04-cuda12.1.0-py310-torch2.3.0-tf2.16.1-1.16.0 /bin/bash 此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”
ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352