23 lines
617 B
Text
23 lines
617 B
Text
|
ARG BASE_IMAGE=myelin-node:latest
|
||
|
|
FROM ${BASE_IMAGE}
|
||
|
|
|
||
|
|
ARG TINYGRAD_VERSION=0.12.0
|
||
|
|
RUN apt-get update && \
|
||
|
|
apt-get install -y --no-install-recommends python3-pip && \
|
||
|
|
python3 -m pip install \
|
||
|
|
--break-system-packages \
|
||
|
|
--no-cache-dir \
|
||
|
|
"tinygrad==${TINYGRAD_VERSION}" && \
|
||
|
|
apt-get purge -y --auto-remove python3-pip && \
|
||
|
|
apt-get clean && \
|
||
|
|
rm -rf /var/lib/apt/lists/* \
|
||
|
|
/var/cache/apt/archives/* \
|
||
|
|
/var/cache/apt/*.bin \
|
||
|
|
/var/log/apt/* \
|
||
|
|
/var/log/dpkg.log \
|
||
|
|
/tmp/* \
|
||
|
|
/var/tmp/*
|
||
|
|
|
||
|
|
ENV DEV=CUDA \
|
||
|
|
CUDA_PTX=1
|