Update .gitlab-ci.yml

.gitlab-ci.yml

@@ -20,8 +20,8 @@ before_script:
 build_pytorch:
   stage: build-pytorch
   script:
-    - docker pull $CONTAINER_JUPYTER_IMAGE || true
-    - docker build -f Dockerfile --pull --cache-from $CONTAINER_JUPYTER_IMAGE --tag $CONTAINER_PYTORCH_IMAGE .
+    - docker pull $CONTAINER_PYTORCH_IMAGE || true
+    - docker build -f Dockerfile --pull --cache-from $CONTAINER_PYTORCH_IMAGE --tag $CONTAINER_PYTORCH_IMAGE .
     - docker push $CONTAINER_PYTORCH_IMAGE