时间戳

首页 > 解决方案 > 如何使用 Docker 构建 YOLACT++？

docker - 如何使用 Docker 构建 YOLACT++？

问题描述

我必须在 docker 环境中构建 yolact++（我使用的是 sagemaker 笔记本）。像这样

ARG PYTORCH="1.3"
ARG CUDA="10.1"
ARG CUDNN="7"
 
FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel

我想运行这个

COPY yolact/external/DCNv2/setup.py /opt/ml/code/external/DCNv2/setup.py
RUN cd /opt/ml/code/external/DCNv2 && \
python setup.py build develop

但我得到了这个错误：

No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda'
Traceback (most recent call last):
File "setup.py", line 64, in <module>
ext_modules=get_extensions(),
File "setup.py", line 41, in get_extensions
raise NotImplementedError('Cuda is not available')
NotImplementedError: Cuda is not available

但是环境支持CUDA。任何人都知道问题出在哪里？

谢谢你。

标签： dockerpytorchdockerfileamazon-sagemaker

解决方案

解决方案 ：

我编辑 /etc/docker/daemon.json 的内容：

{
"runtimes": {
    "nvidia": {
        "path": "/usr/bin/nvidia-container-runtime",
        "runtimeArgs": []
     } 
},
"default-runtime": "nvidia" 
}

然后我重新启动docker守护进程：

sudo system restart docker

它解决了我的问题。

推荐阅读