thanks. I’ve also prepared an easy to use docker setup with gpu support if anyone’s interested. Posted it here.