How to build YOLACT++ using Docker?

Question

I have to build yolact++ in docker enviromment (i'm using sagemaker notebook). Like this

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

And i want to run this

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

But i got this error :

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

But the enviromment supports CUDA. Anyone have an idea where is the problem ?

Thank you.

The error message indicate CUDA is not considered installed by the python compiler. Did you try to check the presence (and the path) of the CUDA installation ? — XouDo
– XouDo, Commented May 18, 2021 at 8:07
I cheched if cuda is available it returns False, but when i check the version of cuda is returns me 10.1, i set CUDA_PATH to CUDA_HOME="/usr/local/cuda-10.1". — Keytich
– Keytich, Commented May 18, 2021 at 9:17
Ok so if it still says using CUDA_HOME='/usr/local/cuda' then your setting of CUDA_HOME does not work. — XouDo
– XouDo, Commented May 18, 2021 at 9:25

Dharman · Accepted Answer · 2021-05-20 14:04:39Z

3

SOLUTION :

i edit the /etc/docker/daemon.json with content:

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

Then i Restart docker daemon:

sudo system restart docker

it solved my problem.

edited May 20, 2021 at 14:04

Dharman♦

33.9k27 gold badges106 silver badges157 bronze badges

answered May 20, 2021 at 13:58

Keytich

415 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

XouDo · Accepted Answer · 2021-05-18 09:46:33Z

0

You should try setting your CUDA_HOME variable in your dockerfile like this :

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 see if the python compiler sees it then.

answered May 18, 2021 at 9:46

XouDo

1,0441 gold badge14 silver badges22 bronze badges

2 Comments

Keytich Over a year ago

it still says : No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda-10.1'

XouDo Over a year ago

So it's not a problem with the path. You said cuda.is_available() returns false... then you have to figure out why. Maybe you can check stackoverflow.com/questions/48152674/…

Collectives™ on Stack Overflow

How to build YOLACT++ using Docker?

2 Answers 2

Comments

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related