mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-02-03 17:55:21 +03:00
43 lines
1.4 KiB
Bash
Executable File
43 lines
1.4 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -euxo pipefail
|
|
|
|
echo "${CUDA_LIB_VERSION:=11.7}"
|
|
echo "${CUDA_PKG_VERSION:=11-7}"
|
|
echo "${CUDNN_VERSION:=8.6.0.84}"
|
|
echo "${TENSORRT_VERSION:=8.4.1}"
|
|
|
|
# Add NVidia Repo
|
|
apt-get -qq update && apt-get install -y --no-install-recommends software-properties-common
|
|
# wget -q -O - https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/3bf863cc.pub | apt-key add -
|
|
# echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/ /" | tee /etc/apt/sources.list.d/nvidia-cuda.list
|
|
wget https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.0-1_all.deb
|
|
dpkg -i cuda-keyring_1.0-1_all.deb
|
|
add-apt-repository contrib
|
|
apt-get -qq update
|
|
|
|
# CUDA Base
|
|
apt-get install -y --no-install-recommends \
|
|
cuda-cudart-${CUDA_PKG_VERSION} \
|
|
cuda-compat-${CUDA_PKG_VERSION} \
|
|
cuda-cupti-${CUDA_PKG_VERSION}
|
|
#ln -s cuda-${CUDA_LIB_VERSION} /usr/local/cuda
|
|
|
|
# CUDA Runtime
|
|
apt-get install -y --no-install-recommends \
|
|
cuda-libraries-${CUDA_PKG_VERSION} \
|
|
libnpp-${CUDA_PKG_VERSION} \
|
|
cuda-nvtx-${CUDA_PKG_VERSION} \
|
|
libcublas-${CUDA_PKG_VERSION}
|
|
|
|
# cuDNN Runtime
|
|
apt-get install -y --no-install-recommends \
|
|
libcudnn8
|
|
|
|
# TensorRT Runtime
|
|
apt-get install -y --no-install-recommends \
|
|
tensorrt-libs python3-libnvinfer
|
|
|
|
# apt-get clean autoclean -y
|
|
# apt-get autoremove --purge -y
|
|
# rm -rf /var/lib/apt/lists/* |