ModelScope 官方提供的 Docker 镜像-阿里云开发者社区

目前 ModelScope 官方提供的 Docker 镜像中，没有包含 CUDA 10 版本的镜像。不过您可以自己构建一个包含 CUDA 10 的镜像，具体步骤如下：

1. 从 Docker Hub 上拉取 nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04 镜像，该镜像包含了 CUDA 10 和 cuDNN 7。

   docker pull nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
   
   

2. 在该镜像的基础上安装 Python3 和 ModelScope。

   FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
   
   RUN apt-get update && \\
       apt-get install -y python3 python3-pip && \\
       pip3 install modelscope
   
   

3. 构建 Docker 镜像。

   docker build -t modelscope-cuda10 .
   
   

4. 运行 Docker 镜像。

   docker run -it modelscope-cuda10