Instruction on Using Nvidia GPU (CUDA) for Computing in Docker
-
Install Nvidia
cuda-drivers(or equivalent) on your Linux machine following instructions at CUDA Downloads. Notice that instead of installingcuda(usingsudo apt-get install cuda), it is suggested that you installcuda-driversonly (usingsudo apt-get install cuda-drivers). This is …