Update container base
This commit is contained in:
@@ -1,12 +1,14 @@
|
||||
# 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
|
||||
ENV USE_CUDA=1
|
||||
ENV BUILD_SOX=1
|
||||
|
||||
# Install system packages to build torchaudio
|
||||
RUN apt update && apt install -y \
|
||||
RUN apt update && \
|
||||
apt upgrade && \
|
||||
apt install -y \
|
||||
libavformat-dev \
|
||||
libavcodec-dev \
|
||||
libavutil-dev \
|
||||
|
||||
Reference in New Issue
Block a user