Update container base
This commit is contained in:
@@ -1,12 +1,14 @@
|
|||||||
# Get cuda version from https://hub.docker.com/r/nvidia/cuda
|
# Get cuda version from https://hub.docker.com/r/nvidia/cuda
|
||||||
FROM nvcr.io/nvidia/pytorch:25.05-py3 AS build-deps
|
FROM nvcr.io/nvidia/pytorch:26.04-py3 AS build-deps
|
||||||
|
|
||||||
# Set ENV
|
# Set ENV
|
||||||
ENV USE_CUDA=1
|
ENV USE_CUDA=1
|
||||||
ENV BUILD_SOX=1
|
ENV BUILD_SOX=1
|
||||||
|
|
||||||
# Install system packages to build torchaudio
|
# Install system packages to build torchaudio
|
||||||
RUN apt update && apt install -y \
|
RUN apt update && \
|
||||||
|
apt upgrade && \
|
||||||
|
apt install -y \
|
||||||
libavformat-dev \
|
libavformat-dev \
|
||||||
libavcodec-dev \
|
libavcodec-dev \
|
||||||
libavutil-dev \
|
libavutil-dev \
|
||||||
|
|||||||
Reference in New Issue
Block a user