Use a docker base image with torch GPU installed

Instead of ultralytics/ultralytics:<version>-python , we should use ultralytics/ultralytics:<version> which includes the torch installation with GPU.