Build with: -
docker build -f dockerfile -t ckai .
Run using: -
docker run --gpus all --device=/dev/video0:/dev/video0 --restart unless-stopped -d -v $HOME:/data/home -p 8888:8888 ckai:latest
Or omit the --device parameter if webcam forwarding is not required.
nvidia-smi
app/package | version |
---|---|
nvidia-smi | 450.51.05 |
driver | 450.51.05 |
CUDA | 11.0 |
nvcc --version
cuda compilation tools, release 10.2, V10.2.89
python package.__version__
app/package | version |
---|---|
tensorflow | 1.13.1 |
torch | 1.5.1 |
keras | 2.3.1 |
cv2 | 4.3.0 |
torch.cuda.is_available() True