cancel
Showing results for 
Show  only  | Search instead for 
Did you mean: 
Highlighted
Participant
Participant
103 Views
Registered: ‎08-15-2020

issue when run docker gpu

Hello, when I am using ./docker_run.sh xilinx/vitis-ai-gpu:latest, it shows that:

docker: Error response from daemon: Unknown runtime specified nvidia.

However, I have followed all the instructions on:

https://github.com/XilinVitis-AIx/

building and installing NVIDIA Docker Runtime, why should this happen?

I am using Ubuntu 18.04 and RTX2080

how can I fix this? 

Thank you very much!

0 Kudos
1 Reply
Highlighted
Xilinx Employee
Xilinx Employee
43 Views
Registered: ‎08-30-2011

With the latest version of docker, the call is slightly different and you may have to modify the call to "docker run" inside the docker_run.sh script. 

Inside the "elif" block, remove

--runtime "nvidia" 

replace it with 

--gpus 'all'

 

Instead of 'all' you can use a specific GPU in your system, it will correspond to the numbering inside nvidia-smi.