docker-cuda-demo https://pytorch.org/docs/stable/notes/cuda.html https://pytorch.org/docs/stable/cuda.html nvidia-smi