目前 ModelScope 官方提供的 Docker 镜像中,没有包含 CUDA 10 版本的镜像。不过您可以自己构建一个包含 CUDA 10 的镜像,具体步骤如下:
从 Docker Hub 上拉取 nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04 镜像,该镜像包含了 CUDA 10 和 cuDNN 7。
docker pull nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
在该镜像的基础上安装 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
构建 Docker 镜像。
docker build -t modelscope-cuda10 .
运行 Docker 镜像。
docker run -it modelscope-cuda10
这样您就可以在包含 CUDA 10 的 Docker 环境中使用 ModelScope 了。当然,如果您需要使用其他版本的 CUDA,可以根据以上步骤自行构建相应的 Docker 镜像。