CODEWITHSUNDEEP
dockeranacondagpu
Vedanta Jha
Vedanta Jha

Reputation: 95

How to make a docker image with Anaconda and be able to use GPU?

I have a GPU, and CUDA driver installed. I am making a docker image using the official anaconda image available at docker hub. When I use the command nvidia-smi inside the docker image, it shows there is no CUDA DRIVER installed.

How should I make a docker image to be able to use Anaconda with GPU?

Upvotes: 1

Views: 2982

Answers (1)

7_R3X
7_R3X

Reputation: 4370

Docker 19.03 has native support for NVIDIA GPUs. Install nvidia-container-runtime and use the docker run --gpus all flag with your Anaconda container. Remember, that you can run Windows containers with GPU acceleration on a Windows host using Docker 19.03, but not a Linux container.

Source: GPU Acceleration in Windows Containers

Upvotes: 2

Related Questions