kwierman / docker_ncsdk

Dockerfile for running software on the Movidius Neural Compute Stick
MIT License
8 stars 5 forks source link

docker_ncsdk

Dockerfile for running software on the Movidius Neural Compute Stick

Dockerhub page: https://hub.docker.com/r/kwierman/movidius/

Installation

docker build -t kwierman/movidius .

Bugfix

If the NCS is plugged into USB3.0, then when it switches modes, the docker image loses the connection. To fix this, add these lines the file: /workspace/ncsdk/api/src/usb_boot.c, inside the function usb_find_device after variable declaration, but before the rest of the code.

        libusb_exit(NULL);
        libusb_init(NULL);

This fix is credited to the discussion here.

Usage

docker run -t -i -v /dev/bus/usb:/dev/bus/usb kwierman/movidius bash

For jupyter-notebook:

docker run -t -i -v /dev/bus/usb:/dev/bus/usb -p 8888:8888 kwierman/movidius jupyter-notebook