ModelScope中,使用社区的cuda docker,在服务器中挂载了gpu,nvidia-smi可以看到当前gpu,但是torch还是使用不了cuda,
我就是这样挂载的gpu,服务器的nvidia驱动和容器的驱动需要一致么
您参考一下这个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开发者联盟群 ①