github/ebook2audiobook
1
1
Fork 0
mirror of https://github.com/DrewThomasson/ebook2audiobook.git synced 2026-01-10 06:18:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
unknown
2025-12-15 20:55:41 -08:00
parent 988abd380f
commit 83dbb37f54
1 changed files with 19 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
Dockerfile
View File

@@ -68,25 +68,6 @@ RUN mkdir -p /usr/lib && \
ln -s /usr/lib64/libX11.so.6 /usr/lib/libX11.so.6 2>/dev/null || true && \
ln -s /usr/lib64/libXext.so.6 /usr/lib/libXext.so.6 2>/dev/null || true && \
ln -s /usr/lib64/libXrender.so.1 /usr/lib/libXrender.so.1 2>/dev/null || true
RUN case "${DEVICE_TAG}" in \
jetson51) \
echo "JetPack 5.1.x → copying CUDA 11.4 libs" && \
mkdir -p /usr/local/cuda-11.4/lib64 && \
( cp -P /usr/lib/aarch64-linux-gnu/libcuda* /usr/local/cuda-11.4/lib64/ 2>/dev/null || true ) && \
( cp -P /usr/lib/aarch64-linux-gnu/libcudart.so.11.0 /usr/local/cuda-11.4/lib64/ 2>/dev/null || true ) && \
( cp -P /usr/lib/aarch64-linux-gnu/libcublas* /usr/local/cuda-11.4/lib64/ 2>/dev/null || true ) && \
( cp -P /usr/lib/aarch64-linux-gnu/libcufft* /usr/local/cuda-11.4/lib64/ 2>/dev/null || true ) && \
( cp -P /usr/lib/aarch64-linux-gnu/libcurand* /usr/local/cuda-11.4/lib64/ 2>/dev/null || true ) && \
( cp -P /usr/lib/aarch64-linux-gnu/libcusparse* /usr/local/cuda-11.4/lib64/ 2>/dev/null || true ) ;; \
jetson60|jetson61) \
echo "JetPack 6.x → no extra CUDA lib copy needed" ;; \
xpu) \
echo "Intel XPU detected — using IPEX" ;; \
rocm*) \
echo "AMD ROCm detected — using ROCm PyTorch" ;; \
*) ;; \
esac
RUN if [ "${DEVICE_TAG}" = "jetson51" ]; then \
echo "LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64" >> /etc/environment; \
@@ -108,6 +89,25 @@ RUN set -eux; \
apt-get purge -y --auto-remove gcc g++ make python3-dev pkg-config git; \
apt-get clean; \
rm -rf /var/lib/apt/lists/*
RUN case "${DEVICE_TAG}" in \
jetson51) \
echo "JetPack 5.1.x → copying CUDA 11.4 libs" && \
mkdir -p /usr/local/cuda-11.4/lib64 && \
( cp -P /usr/lib/aarch64-linux-gnu/libcuda* /usr/local/cuda-11.4/lib64/ 2>/dev/null || true ) && \
( cp -P /usr/lib/aarch64-linux-gnu/libcudart.so.11.0 /usr/local/cuda-11.4/lib64/ 2>/dev/null || true ) && \
( cp -P /usr/lib/aarch64-linux-gnu/libcublas* /usr/local/cuda-11.4/lib64/ 2>/dev/null || true ) && \
( cp -P /usr/lib/aarch64-linux-gnu/libcufft* /usr/local/cuda-11.4/lib64/ 2>/dev/null || true ) && \
( cp -P /usr/lib/aarch64-linux-gnu/libcurand* /usr/local/cuda-11.4/lib64/ 2>/dev/null || true ) && \
( cp -P /usr/lib/aarch64-linux-gnu/libcusparse* /usr/local/cuda-11.4/lib64/ 2>/dev/null || true ) ;; \
jetson60|jetson61) \
echo "JetPack 6.x → no extra CUDA lib copy needed" ;; \
xpu) \
echo "Intel XPU detected — using IPEX" ;; \
rocm*) \
echo "AMD ROCm detected — using ROCm PyTorch" ;; \
*) ;; \
esac
VOLUME /app