mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-02-03 17:55:21 +03:00
54 lines
1.5 KiB
Bash
54 lines
1.5 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
set -euxo pipefail
|
||
|
|
|
||
|
|
"${CUDA_LIB_VERSION:=11.8}"
|
||
|
|
"${CUDA_PKG_VERSION:=11-8}"
|
||
|
|
"${CUDNN_VERSION:=8.6.0.84}"
|
||
|
|
"${TENSORRT_VERSION:=7.2.2}"
|
||
|
|
|
||
|
|
# Add NVidia Repo
|
||
|
|
# 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
|
||
|
|
|
||
|
|
# Utilities
|
||
|
|
apt-get install -y --no-install-recommends \
|
||
|
|
build-essential python3.9-dev python3-pip tar
|
||
|
|
|
||
|
|
python3 -m pip install --upgrade setuptools pip wheel numpy
|
||
|
|
|
||
|
|
# 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
|
||
|
|
|
||
|
|
# Download pycuda source
|
||
|
|
wget https://github.com/inducer/pycuda/archive/refs/tags/v2022.2.tar.gz -O pycuda-v2022.2.tar.gz
|
||
|
|
tar xfz pycuda-v2022.2.tar.gz
|
||
|
|
|
||
|
|
# Build pycuda wheel
|
||
|
|
cd pycuda-v2022.2
|
||
|
|
python setup.py build
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
rm -rf /var/lib/apt/lists/*
|