如何使用Docker构建YOLACT？

Question

我必须在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。有人知道问题出在哪里吗？

谢谢你。

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"

...看看Python编译器是否会看到它。