ModelScope自定义构建 docker镜像
目前 ModelScope 官方提供的 Docker 镜像中,没有包含 CUDA 10 版本的镜像。不过您可以自己构建一个包含 CUDA 10 的镜像,具体步骤如下:
- 从 Docker Hub 上拉取 nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04 镜像,该镜像包含了 CUDA 10 和 cuDNN 7。
- 在该镜像的基础上安装 Python3 和 ModelScope。
- 构建 Docker 镜像。
- 运行 Docker 镜像。
Loading...