entrypoint.sh 4.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. #!/bin/bash -e
  2. trap "echo TRAPed signal" HUP INT QUIT KILL TERM
  3. sudo chown user:user /home/user
  4. echo "user:$PASSWD" | sudo chpasswd
  5. sudo rm -rf /tmp/.X*
  6. sudo ln -snf "/usr/share/zoneinfo/$TZ" /etc/localtime && echo "$TZ" | sudo tee /etc/timezone > /dev/null
  7. sudo ln -snf /dev/ptmx /dev/tty7
  8. sudo /etc/init.d/dbus start
  9. source /opt/gstreamer/gst-env
  10. # Install NVIDIA drivers, including X graphic drivers by omitting --x-{prefix,module-path,library-path,sysconfig-path}
  11. if ! command -v nvidia-xconfig &> /dev/null; then
  12. export DRIVER_VERSION=$(head -n1 </proc/driver/nvidia/version | awk '{print $8}')
  13. cd /tmp
  14. if [ ! -f "/tmp/NVIDIA-Linux-x86_64-$DRIVER_VERSION.run" ]; then
  15. curl -fsL -O "https://us.download.nvidia.com/XFree86/Linux-x86_64/$DRIVER_VERSION/NVIDIA-Linux-x86_64-$DRIVER_VERSION.run" || curl -fsL -O "https://us.download.nvidia.com/tesla/$DRIVER_VERSION/NVIDIA-Linux-x86_64-$DRIVER_VERSION.run" || { echo "Failed NVIDIA GPU driver download. Exiting."; exit 1; }
  16. fi
  17. sudo sh "NVIDIA-Linux-x86_64-$DRIVER_VERSION.run" -x
  18. cd "NVIDIA-Linux-x86_64-$DRIVER_VERSION"
  19. sudo ./nvidia-installer --silent \
  20. --no-kernel-module \
  21. --install-compat32-libs \
  22. --no-nouveau-check \
  23. --no-nvidia-modprobe \
  24. --no-rpms \
  25. --no-backup \
  26. --no-check-for-alternate-installs \
  27. --no-libglx-indirect \
  28. --no-install-libglvnd
  29. sudo rm -rf /tmp/NVIDIA* && cd ~
  30. fi
  31. if grep -Fxq "allowed_users=console" /etc/X11/Xwrapper.config; then
  32. sudo sed -i "s/allowed_users=console/allowed_users=anybody/;$ a needs_root_rights=yes" /etc/X11/Xwrapper.config
  33. fi
  34. if [ -f "/etc/X11/xorg.conf" ]; then
  35. sudo rm "/etc/X11/xorg.conf"
  36. fi
  37. if [ "$NVIDIA_VISIBLE_DEVICES" == "all" ]; then
  38. export GPU_SELECT=$(sudo nvidia-smi --query-gpu=uuid --format=csv | sed -n 2p)
  39. elif [ -z "$NVIDIA_VISIBLE_DEVICES" ]; then
  40. export GPU_SELECT=$(sudo nvidia-smi --query-gpu=uuid --format=csv | sed -n 2p)
  41. else
  42. export GPU_SELECT=$(sudo nvidia-smi --id=$(echo "$NVIDIA_VISIBLE_DEVICES" | cut -d ',' -f1) --query-gpu=uuid --format=csv | sed -n 2p)
  43. if [ -z "$GPU_SELECT" ]; then
  44. export GPU_SELECT=$(sudo nvidia-smi --query-gpu=uuid --format=csv | sed -n 2p)
  45. fi
  46. fi
  47. if [ -z "$GPU_SELECT" ]; then
  48. echo "No NVIDIA GPUs detected. Exiting."
  49. exit 1
  50. fi
  51. HEX_ID=$(sudo nvidia-smi --query-gpu=pci.bus_id --id="$GPU_SELECT" --format=csv | sed -n 2p)
  52. IFS=":." ARR_ID=($HEX_ID)
  53. unset IFS
  54. BUS_ID=PCI:$((16#${ARR_ID[1]})):$((16#${ARR_ID[2]})):$((16#${ARR_ID[3]}))
  55. export MODELINE=$(cvt -r "${SIZEW}" "${SIZEH}" "${REFRESH}" | sed -n 2p)
  56. sudo nvidia-xconfig --virtual="${SIZEW}x${SIZEH}" --depth="$CDEPTH" --mode=$(echo "$MODELINE" | awk '{print $2}' | tr -d '"') --allow-empty-initial-configuration --no-probe-all-gpus --busid="$BUS_ID" --only-one-x-screen --connected-monitor="$VIDEO_PORT"
  57. sudo sed -i '/Driver\s\+"nvidia"/a\ Option "ModeValidation" "NoMaxPClkCheck, NoEdidMaxPClkCheck, NoMaxSizeCheck, NoHorizSyncCheck, NoVertRefreshCheck, NoVirtualSizeCheck, NoExtendedGpuCapabilitiesCheck, NoTotalSizeCheck, NoDualLinkDVICheck, NoDisplayPortBandwidthCheck, AllowNon3DVisionModes, AllowNonHDMI3DModes, AllowNonEdidModes, NoEdidHDMI2Check, AllowDpInterlaced"' /etc/X11/xorg.conf
  58. sudo sed -i '/Section\s\+"Monitor"/a\ '"$MODELINE" /etc/X11/xorg.conf
  59. export DISPLAY=":0"
  60. export __GL_SYNC_TO_VBLANK="0"
  61. Xorg vt7 -novtswitch -sharevts -dpi "${DPI}" +extension "MIT-SHM" "${DISPLAY}" &
  62. # Wait for X11 to start
  63. echo "Waiting for X socket"
  64. until [ -S "/tmp/.X11-unix/X${DISPLAY/:/}" ]; do sleep 1; done
  65. echo "X socket is ready"
  66. if [ "$NOVNC_ENABLE" = "true" ]; then
  67. if [ -n "$NOVNC_VIEWPASS" ]; then export NOVNC_VIEWONLY="-viewpasswd ${NOVNC_VIEWPASS}"; fi
  68. sudo x11vnc -display "${DISPLAY}" -passwd "${BASIC_AUTH_PASSWORD:-$PASSWD}" "${NOVNC_VIEWONLY}" -shared -forever -repeat -xkb -xrandr "resize" -rfbport 5900 &
  69. /opt/noVNC/utils/novnc_proxy --vnc localhost:5900 --listen 8080 --heartbeat 10 &
  70. fi
  71. # Add custom processes below this section or within `supervisord.conf`
  72. mate-session &
  73. echo "Session Running. Press [Return] to exit."
  74. read