diff --git a/.vib/pytorch/goss/goss.yaml b/.vib/pytorch/goss/goss.yaml
index b148ac32b4..5598092b2d 100644
--- a/.vib/pytorch/goss/goss.yaml
+++ b/.vib/pytorch/goss/goss.yaml
@@ -11,6 +11,7 @@ command:
   run-git-example:
     exec: python ./pytorch/goss/testfiles/polynomial_tensor.py
     exit-status: 0
+    timeout: 20000
   {{- $uid := .Vars.containerSecurityContext.runAsUser }}
   {{- $gid := .Vars.podSecurityContext.fsGroup }}
   check-user-info:
diff --git a/bitnami/pytorch/Chart.yaml b/bitnami/pytorch/Chart.yaml
index e431b1122a..95d4b426a0 100644
--- a/bitnami/pytorch/Chart.yaml
+++ b/bitnami/pytorch/Chart.yaml
@@ -33,4 +33,4 @@ maintainers:
 name: pytorch
 sources:
 - https://github.com/bitnami/charts/tree/main/bitnami/pytorch
-version: 4.0.0
+version: 4.0.1
diff --git a/bitnami/pytorch/README.md b/bitnami/pytorch/README.md
index d503e656ae..68b60aa350 100644
--- a/bitnami/pytorch/README.md
+++ b/bitnami/pytorch/README.md
@@ -141,19 +141,19 @@ You can enable this initContainer by setting `volumePermissions.enabled` to `tru
 | `livenessProbe.enabled`                             | Enable livenessProbe                                                                                                                                                                                       | `true`                    |
 | `livenessProbe.initialDelaySeconds`                 | Initial delay seconds for livenessProbe                                                                                                                                                                    | `5`                       |
 | `livenessProbe.periodSeconds`                       | Period seconds for livenessProbe                                                                                                                                                                           | `5`                       |
-| `livenessProbe.timeoutSeconds`                      | Timeout seconds for livenessProbe                                                                                                                                                                          | `5`                       |
+| `livenessProbe.timeoutSeconds`                      | Timeout seconds for livenessProbe                                                                                                                                                                          | `20`                      |
 | `livenessProbe.failureThreshold`                    | Failure threshold for livenessProbe                                                                                                                                                                        | `5`                       |
 | `livenessProbe.successThreshold`                    | Success threshold for livenessProbe                                                                                                                                                                        | `1`                       |
 | `readinessProbe.enabled`                            | Enable readinessProbe                                                                                                                                                                                      | `true`                    |
 | `readinessProbe.initialDelaySeconds`                | Initial delay seconds for readinessProbe                                                                                                                                                                   | `5`                       |
 | `readinessProbe.periodSeconds`                      | Period seconds for readinessProbe                                                                                                                                                                          | `5`                       |
-| `readinessProbe.timeoutSeconds`                     | Timeout seconds for readinessProbe                                                                                                                                                                         | `3`                       |
+| `readinessProbe.timeoutSeconds`                     | Timeout seconds for readinessProbe                                                                                                                                                                         | `20`                      |
 | `readinessProbe.failureThreshold`                   | Failure threshold for readinessProbe                                                                                                                                                                       | `5`                       |
 | `readinessProbe.successThreshold`                   | Success threshold for readinessProbe                                                                                                                                                                       | `1`                       |
 | `startupProbe.enabled`                              | Enable startupProbe                                                                                                                                                                                        | `true`                    |
 | `startupProbe.initialDelaySeconds`                  | Initial delay seconds for startupProbe                                                                                                                                                                     | `5`                       |
 | `startupProbe.periodSeconds`                        | Period seconds for startupProbe                                                                                                                                                                            | `5`                       |
-| `startupProbe.timeoutSeconds`                       | Timeout seconds for startupProbe                                                                                                                                                                           | `3`                       |
+| `startupProbe.timeoutSeconds`                       | Timeout seconds for startupProbe                                                                                                                                                                           | `20`                      |
 | `startupProbe.failureThreshold`                     | Failure threshold for startupProbe                                                                                                                                                                         | `5`                       |
 | `startupProbe.successThreshold`                     | Success threshold for startupProbe                                                                                                                                                                         | `1`                       |
 | `customLivenessProbe`                               | Custom livenessProbe that overrides the default one                                                                                                                                                        | `{}`                      |
diff --git a/bitnami/pytorch/templates/deployment.yaml b/bitnami/pytorch/templates/deployment.yaml
index 069766cd38..77e31073d4 100644
--- a/bitnami/pytorch/templates/deployment.yaml
+++ b/bitnami/pytorch/templates/deployment.yaml
@@ -169,7 +169,7 @@ spec:
               command:
                 - python
                 - -c
-                - import torch; torch.__version__
+                - import torch; print(torch.__version__)
           {{- end }}
           {{- if .Values.customReadinessProbe }}
           readinessProbe: {{- include "common.tplvalues.render" (dict "value" .Values.customReadinessProbe "context" $) | nindent 12 }}
@@ -179,7 +179,7 @@ spec:
               command:
                 - python
                 - -c
-                - import torch; torch.__version__
+                - import torch; print(torch.__version__)
           {{- end }}
           {{- if .Values.customStartupProbe }}
           startupProbe: {{- include "common.tplvalues.render" (dict "value" .Values.customStartupProbe "context" $) | nindent 12 }}
@@ -189,7 +189,7 @@ spec:
               command:
                 - python
                 - -c
-                - import torch; torch.__version__
+                - import torch; print(torch.__version__)
           {{- end }}
           {{- end }}
           {{- if .Values.lifecycleHooks }}
diff --git a/bitnami/pytorch/templates/statefulset.yaml b/bitnami/pytorch/templates/statefulset.yaml
index f1ddabb2a8..1508d3eae2 100644
--- a/bitnami/pytorch/templates/statefulset.yaml
+++ b/bitnami/pytorch/templates/statefulset.yaml
@@ -177,7 +177,7 @@ spec:
               command:
                 - python
                 - -c
-                - import torch; torch.__version__
+                - import torch; print(torch.__version__)
           {{- end }}
           {{- if .Values.customReadinessProbe }}
           readinessProbe: {{- include "common.tplvalues.render" (dict "value" .Values.customReadinessProbe "context" $) | nindent 12 }}
@@ -187,7 +187,7 @@ spec:
               command:
                 - python
                 - -c
-                - import torch; torch.__version__
+                - import torch; print(torch.__version__)
           {{- end }}
           {{- if .Values.customStartupProbe }}
           startupProbe: {{- include "common.tplvalues.render" (dict "value" .Values.customStartupProbe "context" $) | nindent 12 }}
@@ -197,7 +197,7 @@ spec:
               command:
                 - python
                 - -c
-                - import torch; torch.__version__
+                - import torch; print(torch.__version__)
           {{- end }}
           {{- end }}
           {{- if .Values.lifecycleHooks }}
diff --git a/bitnami/pytorch/values.yaml b/bitnami/pytorch/values.yaml
index 28c844a866..bd29595575 100644
--- a/bitnami/pytorch/values.yaml
+++ b/bitnami/pytorch/values.yaml
@@ -117,7 +117,7 @@ livenessProbe:
   enabled: true
   initialDelaySeconds: 5
   periodSeconds: 5
-  timeoutSeconds: 5
+  timeoutSeconds: 20
   successThreshold: 1
   failureThreshold: 5
 ## Configure extra options for readiness probe
@@ -133,7 +133,7 @@ readinessProbe:
   enabled: true
   initialDelaySeconds: 5
   periodSeconds: 5
-  timeoutSeconds: 3
+  timeoutSeconds: 20
   successThreshold: 1
   failureThreshold: 5
 ## Configure extra options for startup probe
@@ -149,7 +149,7 @@ startupProbe:
   enabled: true
   initialDelaySeconds: 5
   periodSeconds: 5
-  timeoutSeconds: 3
+  timeoutSeconds: 20
   successThreshold: 1
   failureThreshold: 5
 ## @param customLivenessProbe Custom livenessProbe that overrides the default one