TAO 5.0 Release - PyTorch

Author: Arun-George-Zachariah

release/docker/Dockerfile

@@ -52,17 +52,8 @@ RUN rm -rf /usr/local/lib/python3.8/dist-packages/mpmath-1.2.1.dist-info
 RUN rm -rf /opt/nvidia/entrypoint.d/*.txt
 COPY release/docker/entrypoint.d/* /opt/nvidia/entrypoint.d/
 
-# Copying tlt-converter and building it in the docker.
-ENV TRT_LIB_PATH="/usr/lib/x86_64-linux-gnu"
-ENV TRT_INCLUDE_PATH="/usr/include/x86_64-linux-gnu"
-COPY tao-converter /opt/nvidia/tools
-WORKDIR /opt/nvidia/tools
-RUN make \
-    && rm -rf Makefile *.cc *.h *.o
-RUN ln -s tao-converter converter
-
 ENV NVIDIA_PRODUCT_NAME "TAO Toolkit"
-ENV TAO_TOOLKIT_VERSION="4.0.0"
+ENV TAO_TOOLKIT_VERSION="5.0.0"
 ENV NVIDIA_TAO_TOOLKIT_VERSION="${TAO_TOOLKIT_VERSION}-PyTorch"
 
 ENV NVM_DIR=/usr/local/nvm

release/docker/build_wheel.sh

@@ -1,15 +1,10 @@
 #!/bin/bash
-# Description: Script responsible for generation of an obf_src wheel using pyarmor package.
+# Description: Script responsible for generation of an src wheel
 
 # Setting the repo root for local build environment or CI.
 [[ -z "${WORKSPACE}" ]] && REPO_ROOT='/tao-pt' || REPO_ROOT="${WORKSPACE}"
 echo "Building from ${REPO_ROOT}"
 
-echo "Installing required packages"
-pip install pyarmor pyinstaller pybind11
-echo "Registering pyarmor"
-pyarmor register ${REPO_ROOT}/release/docker/pyarmor-regfile-1219.zip || exit $?
-
 echo "Clearing build and dists"
 python ${REPO_ROOT}/setup.py clean --all
 rm -rf dist/*
@@ -20,25 +15,20 @@ find . | grep -E "(__pycache__|\.pyc|\.pyo$)" | xargs rm -rf
 mkdir /dist
 cp -r ${REPO_ROOT}/nvidia_tao_pytorch/* /dist/
 
-echo "Obfuscating the code using pyarmor"
-cd ${REPO_ROOT}
-python -c "from release.python.utils import encrypt_source_code; encrypt_source_code.encrypt_files('${REPO_ROOT}/nvidia_tao_pytorch')"
-
 echo "Migrating codebase"
 # Move sources to orig_src
 rm -rf /orig_src
 mkdir /orig_src
 mv ${REPO_ROOT}/nvidia_tao_pytorch/* /orig_src/
 
-# Move obf_src files to src
+# Move files to src
 mv /dist/* ${REPO_ROOT}/nvidia_tao_pytorch/
-mv ${REPO_ROOT}/nvidia_tao_pytorch/pytransform_vax_001219 ${REPO_ROOT}/
 
 echo "Building bdist wheel"
 python setup.py bdist_wheel || exit $?
 
 echo "Restoring the original project structure"
-# Move the obf_src files.
+# Move the files.
 rm -rf ${REPO_ROOT}/nvidia_tao_pytorch/*
 
 # Move back the original files
@@ -47,5 +37,3 @@ mv /orig_src/* ${REPO_ROOT}/nvidia_tao_pytorch/
 # Remove the tmp folders.
 rm -rf /dist
 rm -rf /orig_src
-rm -rf /obf_src
-rm -rf ${REPO_ROOT}/pytransform_vax_001219