Using PyTorch Docker Image from Nvidia (No CUDA installation required)

Docker is a more advanced approach and there’s alway issues with Docker details and how it interacts with CUDA. So unless you’re prepared to spend some time debugging and are interested in learning about this in particular, I’d suggest avoiding Docker for deep learning.