Copernicus-57B / engine /Dockerfile.conv

Add files using upload-large-folder tool

f0304c9 verified 9 days ago

471 Bytes

FROM reax13-base
RUN pip install --no-cache-dir 'ninja==1.13.2' 'nvidia-cuda-cccl==13.0.85'
RUN CAUSAL_CONV1D_FORCE_BUILD=TRUE MAX_JOBS=4 CUDA_HOME=/usr/local/cuda TORCH_CUDA_ARCH_LIST=12.0 pip install --no-cache-dir --no-build-isolation --no-deps 'causal-conv1d==1.7.0'
RUN python -c 'import causal_conv1d, torch; print(causal_conv1d.__version__, torch.__version__)'

RUN pip install --no-cache-dir 'httpx==0.28.1'

COPY . /engine
ENV PYTHONPATH=/engine
WORKDIR /engine

	FROM reax13-base
	RUN pip install --no-cache-dir 'ninja==1.13.2' 'nvidia-cuda-cccl==13.0.85'
	RUN CAUSAL_CONV1D_FORCE_BUILD=TRUE MAX_JOBS=4 CUDA_HOME=/usr/local/cuda TORCH_CUDA_ARCH_LIST=12.0 pip install --no-cache-dir --no-build-isolation --no-deps 'causal-conv1d==1.7.0'
	RUN python -c 'import causal_conv1d, torch; print(causal_conv1d.__version__, torch.__version__)'

	RUN pip install --no-cache-dir 'httpx==0.28.1'

	COPY . /engine
	ENV PYTHONPATH=/engine
	WORKDIR /engine