upgrade to torch 2.0.1

.github/workflows/docker_rdma.yml

@@ -30,4 +30,4 @@ jobs:
         with:
           context: ./docker/rdma/
           push: true
-          tags: dptechnology/unicore:${{ github.ref_name }}-pytorch2.0.0-cuda11.7-rdma
+          tags: dptechnology/unicore:${{ github.ref_name }}-pytorch2.0.1-cuda11.7-rdma

.github/workflows/docker_rdma_latest.yml

@@ -30,4 +30,4 @@ jobs:
         with:
           context: ./docker/rdma/
           push: true
-          tags: dptechnology/unicore:latest-pytorch2.0.0-cuda11.7-rdma
+          tags: dptechnology/unicore:latest-pytorch2.0.1-cuda11.7-rdma

docker/rdma/Dockerfile

@@ -70,7 +70,7 @@ RUN conda install -y ninja typing && \
 
 RUN pip3 install --upgrade sentry-sdk requests
 
-RUN pip3 install torch==2.0.0+cu117  --index-url https://download.pytorch.org/whl/cu117 && rm -rf ~/.cache/pip
+RUN pip3 install torch==2.0.1+cu117  --index-url https://download.pytorch.org/whl/cu117 && rm -rf ~/.cache/pip
 
 RUN cd /tmp && \
     git clone https://github.com/dptech-corp/Uni-Core && \