如何使用Docker构建YOLACT++？

Question

我必须在码头环境中构建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

但环境支持数据自动化系统。有人知道问题出在哪里吗？

谢谢。

Answer 1

解决办法：

我用以下内容编辑/etc/docker/daemon.json：

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

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

sudo system restart docker

它解决了我的问题。

Answer 2

您应该尝试在您的dockerfile中设置CUDA_HOME变量，如下所示：

ARG PYTORCH="1.3"
ARG CUDA="10.1"
ARG CUDNN="7"

FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
ARG CUDA_HOME="/usr/local/cuda-10.1"

...and查看python编译器是否会看到它。