Docker Arguments for Xavier (without dGPU) and Pegasus (with dGPU)

run.bash

#!/bin/bash

# For both Xavier (without dGPU) and Pegasus (with dGPU))
NV_DEV="--device /dev/nvidiactl --device /dev/nvmap"
NV_DGPU="--device /dev/nvgpu-pci"
# To Do: for xavier (without dGPU)

NV_IGPU=$(\ls /dev/nvhost* | xargs -I{} echo '--device {}')
NV_LIBS="-v /usr/lib/libcuda.so.1:/usr/lib/libcuda.so.1 $(\ls /usr/lib/libnv* | xargs -I{} echo '-v {}:{}')"
GL_LIBS="$(\ls /usr/lib/lib*GL* | xargs -I{} echo '-v {}:{}')"
CUDA_DIR="-v /usr/local/cuda-10.0:/usr/local/cuda-10.0"
#X_AUTH="/tmp/.docker.xauth"
X_SOCK="/tmp/.X11-unix"
X_ARG="--net=host -v ${X_SOCK}:${X_SOCK}"
DISP_ARG="-e DISPLAY -e QT_X11_NO_MITSHM=1"
#SSD_MOUNT="$(\ls /media/nvidia | xargs -I{} echo '-v /media/nvidia/{}:/media/nvidia/{}')"
DOCKER_ARGUMENTS="${NV_DEV} ${NV_IGPU} ${NV_DGPU} ${NV_LIBS} ${GL_LIBS} ${CUDA_DIR} ${X_ARG} ${DISP_ARG}"

#DOCKER_ARGUMENTS="${NV_DEV} ${NV_IGPU} ${NV_DGPU} ${NV_LIBS} ${GL_LIBS} ${CUDA_DIR} ${X_ARG} ${DISP_ARG} ${SSD_MOUNT}"


IMAGE_NAME=xenial-pegasus
# allows x access for local:root
xhost +local:root
xhost +
echo "docker run/ -it --rm --privileged ${DOCKER_ARGUMENTS} -v /home/nvidia:/home/nvidia/shared ${HOST_SHARE} ${WORKDIR} ${IMAGE_NAME} bash"
docker run -it --rm --privileged ${DOCKER_ARGUMENTS} -v /home/nvidia:/home/nvidia/shared ${HOST_SHARE} ${WORKDIR} ${IMAGE_NAME} bash