-
Notifications
You must be signed in to change notification settings - Fork 3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
8 changed files
with
95 additions
and
155 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Submodule containers
added at
ddb1f2
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,53 +1,62 @@ | ||
#!/usr/bin/env bash | ||
|
||
# find L4T_VERSION | ||
source tools/l4t-version.sh | ||
source docker/containers/scripts/l4t_version.sh | ||
|
||
# local container:tag name | ||
CONTAINER_IMAGE="jetson-inference:r$L4T_VERSION" | ||
if [ $ARCH = "aarch64" ]; then | ||
# local container:tag name | ||
CONTAINER_IMAGE="jetson-inference:r$L4T_VERSION" | ||
|
||
# incompatible L4T version | ||
function version_error() | ||
{ | ||
echo "cannot find compatible jetson-inference docker container for L4T R$L4T_VERSION" | ||
echo "please upgrade to the latest JetPack, or build jetson-inference natively from source" | ||
exit 1 | ||
} | ||
# incompatible L4T version | ||
function version_error() | ||
{ | ||
echo "cannot find compatible jetson-inference docker container for L4T R$L4T_VERSION" | ||
echo "please upgrade to the latest JetPack, or build jetson-inference natively from source" | ||
exit 1 | ||
} | ||
|
||
# get remote container URL | ||
if [ $L4T_RELEASE -eq 32 ]; then | ||
if [ $L4T_REVISION_MAJOR -eq 4 ]; then | ||
if [ $L4T_REVISION_MINOR -gt 4 ]; then | ||
CONTAINER_REMOTE_IMAGE="nvcr.io/ea-linux4tegra/$CONTAINER_IMAGE" | ||
elif [ $L4T_REVISION_MINOR -ge 3 ]; then | ||
# get remote container URL | ||
if [ $L4T_RELEASE -eq 32 ]; then | ||
if [ $L4T_REVISION_MAJOR -eq 4 ]; then | ||
if [ $L4T_REVISION_MINOR -gt 4 ]; then | ||
CONTAINER_REMOTE_IMAGE="nvcr.io/ea-linux4tegra/$CONTAINER_IMAGE" | ||
elif [ $L4T_REVISION_MINOR -ge 3 ]; then | ||
CONTAINER_REMOTE_IMAGE="dustynv/$CONTAINER_IMAGE" | ||
else | ||
version_error | ||
fi | ||
elif [ $L4T_REVISION_MAJOR -eq 5 ]; then | ||
if [ $L4T_REVISION_MINOR -eq 0 ]; then | ||
CONTAINER_REMOTE_IMAGE="dustynv/$CONTAINER_IMAGE" | ||
elif [ $L4T_REVISION_MINOR -eq 1 ] || [ $L4T_REVISION_MINOR -eq 2 ]; then | ||
# L4T R32.5.1 / R32.5.2 runs the R32.5.0 container | ||
CONTAINER_IMAGE="jetson-inference:r32.5.0" | ||
CONTAINER_REMOTE_IMAGE="dustynv/$CONTAINER_IMAGE" | ||
else | ||
CONTAINER_REMOTE_IMAGE="nvcr.io/ea-linux4tegra/$CONTAINER_IMAGE" | ||
fi | ||
elif [ $L4T_REVISION_MAJOR -gt 5 ]; then | ||
CONTAINER_REMOTE_IMAGE="dustynv/$CONTAINER_IMAGE" | ||
else | ||
version_error | ||
fi | ||
elif [ $L4T_REVISION_MAJOR -eq 5 ]; then | ||
if [ $L4T_REVISION_MINOR -eq 0 ]; then | ||
CONTAINER_REMOTE_IMAGE="dustynv/$CONTAINER_IMAGE" | ||
elif [ $L4T_REVISION_MINOR -eq 1 ] || [ $L4T_REVISION_MINOR -eq 2 ]; then | ||
# L4T R32.5.1 / R32.5.2 runs the R32.5.0 container | ||
CONTAINER_IMAGE="jetson-inference:r32.5.0" | ||
CONTAINER_REMOTE_IMAGE="dustynv/$CONTAINER_IMAGE" | ||
else | ||
CONTAINER_REMOTE_IMAGE="nvcr.io/ea-linux4tegra/$CONTAINER_IMAGE" | ||
fi | ||
elif [ $L4T_REVISION_MAJOR -gt 5 ]; then | ||
elif [ $L4T_RELEASE -eq 34 ]; then | ||
CONTAINER_REMOTE_IMAGE="dustynv/$CONTAINER_IMAGE" | ||
else | ||
version_error | ||
fi | ||
elif [ $L4T_RELEASE -eq 34 ]; then | ||
CONTAINER_REMOTE_IMAGE="dustynv/$CONTAINER_IMAGE" | ||
else | ||
version_error | ||
elif [ $ARCH = "x86_64" ]; then | ||
# TODO: add logic here for getting the latest release | ||
CONTAINER_IMAGE="jetson-inference:22.06" | ||
fi | ||
|
||
# TAG is always the local container name | ||
# CONTAINER_REMOTE_IMAGE is always the container name in NGC/dockerhub | ||
# CONTAINER_IMAGE is either local container name (if it exists on the system) or the remote image name | ||
TAG=$CONTAINER_IMAGE | ||
|
||
# check for local image | ||
if [[ "$(sudo docker images -q $CONTAINER_IMAGE 2> /dev/null)" == "" ]]; then | ||
CONTAINER_IMAGE=$CONTAINER_REMOTE_IMAGE | ||
fi | ||
|
||
|
This file was deleted.
Oops, something went wrong.