can we use --gpu in docker directly ie without Nvidia-docker?

rocker-org / ml

experimental machine learning container

GNU General Public License v2.0

50 stars 13 forks source link

can we use --gpu in docker directly ie without Nvidia-docker? #30

Closed stnava closed 4 years ago

stnava commented 4 years ago

is nvidia-docker is necessary anymore. With 19.03 and older, can you do (after installing nvidia-docker-cli and nvidia-docker-toolkit) pass --gpus= parameter to docker ?

cboettig commented 4 years ago

Correct. I regularly run the rocker/ml images with standard docker run and the --gpus= flag instead.

(Though as you may know docker-compose no longer works with gpu option)

stnava commented 4 years ago

followed these steps : https://github.com/NVIDIA/nvidia-docker/issues/1243

and it worked.

thank you!

stnava commented 4 years ago

also needed

export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}