Update PyTorch to 2.1.0 release version

Also update to latest UBI image version

Author: njhill

Dockerfile

@@ -1,9 +1,9 @@
 ## Global Args #################################################################
-ARG BASE_UBI_IMAGE_TAG=9.2-755
+ARG BASE_UBI_IMAGE_TAG=9.2-755.1696515532
 ARG PROTOC_VERSION=24.3
-#ARG PYTORCH_INDEX="https://download.pytorch.org/whl"
-ARG PYTORCH_INDEX="https://download.pytorch.org/whl/nightly"
-ARG PYTORCH_VERSION=2.1.0.dev20230904
+ARG PYTORCH_INDEX="https://download.pytorch.org/whl"
+#ARG PYTORCH_INDEX="https://download.pytorch.org/whl/nightly"
+ARG PYTORCH_VERSION=2.1.0
 
 ## Base Layer ##################################################################
 FROM registry.access.redhat.com/ubi9/ubi:${BASE_UBI_IMAGE_TAG} as base