Can you add CUDA to a docker container?

• nvidia-container-runtime

  2 1,089 0.0 Makefile

  Discontinued NVIDIA container runtime

  

Yes, you can, actually already exist images with Cuda installed https://hub.docker.com/r/nvidia/cuda . To be able to use the GPU device within the docker container you need to install `nvidia-container-runtime` https://github.com/NVIDIA/nvidia-container-runtime.

• nvidia-docker

  53 16,998 0.0 Makefile

  Discontinued Build and run Docker containers leveraging NVIDIA GPUs

  

Yes it is possible: you have to install NVIDIA Container Toolkit. You can follow this guide. Once you have this toolkit you can create your own image or pull some offical image with pytorch or your Cuda software (your docker image must have the same Cuda version that you have installed in your host machine)

• InfluxDB

  www.influxdata.com sponsored

  Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

  

• container-images

  9 - -

You can use the cuda dockerfile as reference: https://gitlab.com/nvidia/container-images/cuda/-/blob/master/Dockerfile

Suggest a related project