Comment by ydau

Lambda Stack is an alternative to Pop!_OS's tensorman, which doesn't use containers:

https://lambdalabs.com/lambda-stack-deep-learning-software

This is a one-line apt/aptitude installation for TensorFlow, PyTorch, CUDA, cuDNN, etc. When NVIDIA releases a new version of CUDA, you can simply apt-get upgrade to the latest version.

Disclosure: I work for Lambda Labs.