@@ -34,28 +34,29 @@ PROJECT_ROOT="$( cd ${SCRIPTS_DIR}/.. && pwd )"
3434: ${NVIDIA_DOCKER_ROOT:= ${PROJECT_ROOT} / third_party/ nvidia-docker}
3535
3636# Get version for libnvidia-container
37- libnvidia_container_version =$( grep " #define NVC_VERSION" ${LIBNVIDIA_CONTAINER_ROOT} /src/nvc.h \
37+ libnvidia_container_version_tag =$( grep " #define NVC_VERSION" ${LIBNVIDIA_CONTAINER_ROOT} /src/nvc.h \
3838 | sed -e ' s/#define NVC_VERSION[[:space:]]"\(.*\)"/\1/' )
3939
4040# Get version for nvidia-container-toolit
4141nvidia_container_toolkit_version=$( grep -m 1 " ^LIB_VERSION := " ${NVIDIA_CONTAINER_TOOLKIT_ROOT} /Makefile | sed -e ' s/LIB_VERSION :=[[:space:]]\(.*\)[[:space:]]*/\1/' )
4242nvidia_container_toolkit_tag=$( grep -m 1 " ^LIB_TAG .= " ${NVIDIA_CONTAINER_TOOLKIT_ROOT} /Makefile | sed -e ' s/LIB_TAG .=[[:space:]]\(.*\)[[:space:]]*/\1/' )
43- nvidia_container_toolkit_version =" ${nvidia_container_toolkit_version}${nvidia_container_toolkit_tag: +~${nvidia_container_toolkit_tag} } "
43+ nvidia_container_toolkit_version_tag =" ${nvidia_container_toolkit_version}${nvidia_container_toolkit_tag: +~${nvidia_container_toolkit_tag} } "
4444
4545# Get version for nvidia-container-runtime
4646nvidia_container_runtime_version=$( grep -m 1 " ^LIB_VERSION := " ${NVIDIA_CONTAINER_RUNTIME_ROOT} /Makefile | sed -e ' s/LIB_VERSION :=[[:space:]]\(.*\)[[:space:]]*/\1/' )
4747nvidia_container_runtime_tag=$( grep -m 1 " ^LIB_TAG .= " ${NVIDIA_CONTAINER_RUNTIME_ROOT} /Makefile | sed -e ' s/LIB_TAG .=[[:space:]]\(.*\)[[:space:]]*/\1/' )
48- nvidia_container_runtime_version =" ${nvidia_container_runtime_version}${nvidia_container_runtime_tag: +~${nvidia_container_runtime_tag} } "
48+ nvidia_container_runtime_version_tag =" ${nvidia_container_runtime_version}${nvidia_container_runtime_tag: +~${nvidia_container_runtime_tag} } "
4949
5050# Get version for nvidia-docker
5151nvidia_docker_version=$( grep -m 1 " ^LIB_VERSION := " ${NVIDIA_DOCKER_ROOT} /Makefile | sed -e ' s/LIB_VERSION :=[[:space:]]\(.*\)[[:space:]]*/\1/' )
5252nvidia_docker_tag=$( grep -m 1 " ^LIB_TAG .= " ${NVIDIA_DOCKER_ROOT} /Makefile | sed -e ' s/LIB_TAG .=[[:space:]]\(.*\)[[:space:]]*/\1/' )
53- nvidia_docker_version =" ${nvidia_docker_version}${nvidia_docker_tag: +~${nvidia_docker_tag} } "
53+ nvidia_docker_version_tag =" ${nvidia_docker_version}${nvidia_docker_tag: +~${nvidia_docker_tag} } "
5454
5555
56- echo " LIBNVIDIA_CONTAINER_VERSION=${libnvidia_container_version } "
56+ echo " LIBNVIDIA_CONTAINER_VERSION=${libnvidia_container_version_tag } "
5757echo " NVIDIA_CONTAINER_TOOLKIT_VERSION=${nvidia_container_toolkit_version} "
58- if [[ " ${libnvidia_container_version} " != " ${nvidia_container_toolkit_version} " ]]; then
58+ echo " NVIDIA_CONTAINER_TOOLKIT_TAG=${nvidia_container_toolkit_tag} "
59+ if [[ " ${libnvidia_container_version_tag} " != " ${nvidia_container_toolkit_version_tag} " ]]; then
5960 >&2 echo " WARNING: The libnvidia-container and nvidia-container-toolkit versions do not match"
6061fi
6162echo " NVIDIA_CONTAINER_RUNTIME_VERSION=${nvidia_container_runtime_version} "
0 commit comments