@kaiseryet
Yeah, I did multiple reinstallations to no avail, but as I said, installing the nvidia-container-toolkit fixed the issue because my issue was accessing the GPU through docker. It took me time to figure this one out because torch.version.cuda==10.1 and torch.backends.cudnn.enabled==True led me to believe Docker wasn’t the problem.