skip pytorch

Author: david-tingdahl-nvidia

ci/premerge.py

@@ -92,11 +92,7 @@ def build_args(self) -> List[str]:
         if self.args.gcc_sanitizer == 1:
             cmake_args += ' -DCMAKE_BUILD_TYPE=Debug -DUSE_SANITIZER=yes'
 
-        # Pytorch is not supported on CUDA 13.
-        if self.args.cuda_version == CudaVersion.CUDA_13:
-            cmake_args += ' -DBUILD_PYTORCH_WRAPPER=0'
-            print(
-                '::warning :: Pytorch is not yet supported on CUDA 13. Disabling pytorch wrapper.')
+        cmake_args += ' -DBUILD_PYTORCH_WRAPPER=0'
 
         # Setup args to docker build
         args = [f'CMAKE_ARGS={cmake_args}']

docker/Dockerfile.deps

@@ -29,11 +29,11 @@ RUN apt update && apt-get --no-install-recommends install -y \
 # Note that all python deps are installed inside a venv. This is required by ubuntu >= 24.
 # umask 000 is needed to make the python libs readable by all users.
 RUN umask 000 && python3 -m venv /opt/venv
-RUN umask 000 && . /opt/venv/bin/activate && \
-    python3 -m pip install --ignore-installed --upgrade pip --no-cache-dir && \
-    python3 -m pip install --no-cache-dir \
-    # Pytorch. Note that this version is only compatible with pre-cxx11 ABI binaries. \
-    torch==2.4
+# RUN umask 000 && . /opt/venv/bin/activate && \
+#     python3 -m pip install --ignore-installed --upgrade pip --no-cache-dir && \
+#     python3 -m pip install --no-cache-dir \
+#     # Pytorch. Note that this version is only compatible with pre-cxx11 ABI binaries. \
+#     torch==2.4
 
 # Install python deps.
 RUN umask 000 && . /opt/venv/bin/activate && \