Hocam en sağda ataç icon un orada Docker İmajı verilmiş zaten , bu imajı çekerek kurulumu yapabilirsiniz. Ancak burada önemli bi faktör var GPU gerektiği için NVİDİA yapılandırması yapmanız gerekiyor.
O yüzden python üzerinde normal script gibi çalıştırırken de sorun yaşayabilirsiniz. "nvidia-docker2" yazan yerde verilen link ile o kurulumuda gerçekleştirdikten sonra imajı buradan çekerek çalıştırabilirsiniz.

Müsait olsaydım yardımcı olurdum.