Gašper Spagnolo

I can try and integrate the venv version no problemo:) it looks cleaner yeah:) Thanks for pointing out!

The image downloaded in the Dockerfile mentioned in pr #5 is kinda large (10GB) [nvidia containers list](https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch/tags). But lets use it as it has nvidia kernel options available (other image...

Now it is cleaned up thanks @jonppe