Dockerfile - Images are not up to date and doesn't support NVidia CUDA (Fix inside)
Hi @moeiscool ,
Since the policy change by DockerHub the images are not up to date, I have managed to build them locally and applied change to make them work better than it worked before.
My suggestion is to have the Dockerfile.nvidia as the official one (add changes to the original one) so if the server running the image supports NVidia CUDA, it will be supported out of the box as being supported for InteL GPU.
Exiting Dockerfile
Upgraded base build to NodeJS 16.18 Slim version and allowed working with the pm2.yml from the Docker directory.
First line:
Replaced
FROM node:12.21.0-buster-slim
with
FROM node:16.13-buster-slim
Last line:
Replaced
CMD [ "pm2-docker", "pm2.yml" ]
with
CMD [ "pm2-docker", "/home/Shinobi/Docker/pm2.yml" ]
New Dockerfile.nvidia
Created new Dockerfile to support nvidia image (using it currently)
First line:
Replaced
FROM node:12.21.0-buster-slim
with
FROM nvidia/cuda:11.4.2-cudnn8-runtime-ubuntu20.04
After packages installation at line 23, added NodeJS 16 installation
RUN curl -sL https://deb.nodesource.com/setup_16.x | bash -
RUN apt install nodejs -y
RUN node -v
RUN npm -v
Last line:
Replaced
CMD [ "pm2-docker", "pm2.yml" ]
with
CMD [ "pm2-docker", "/home/Shinobi/Docker/pm2.yml" ]
Docker images for test available at:
eladbar/shinobi:latest / eladbar/shinobi:nvidia
Elad