38 lines
1.1 KiB
Plaintext
38 lines
1.1 KiB
Plaintext
|
FROM docker.io/nvidia/cuda:12.3.2-devel-ubuntu22.04
|
||
|
|
|
||
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
||
|
|
|
||
|
|
# Sometimes you have to get update twice because ?
|
||
|
|
RUN apt-get update
|
||
|
|
RUN apt-get update
|
||
|
|
|
||
|
|
RUN apt-get install -y wget unzip python3-pip vim python3 python3-pip curl
|
||
|
|
|
||
|
|
RUN wget https://go.dev/dl/go1.22.2.linux-amd64.tar.gz
|
||
|
|
RUN tar -xvf go1.22.2.linux-amd64.tar.gz -C /usr/local
|
||
|
|
ENV PATH=$PATH:/usr/local/go/bin
|
||
|
|
ENV GOPATH=/go
|
||
|
|
|
||
|
|
RUN bash -c 'curl -L "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.9.1.tar.gz" | tar -C /usr/local -xz'
|
||
|
|
RUN bash -c 'curl -L "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.15.0.tar.gz" | tar -C /usr/local -xz'
|
||
|
|
RUN ldconfig
|
||
|
|
|
||
|
|
RUN ln -s /usr/bin/python3 /usr/bin/python
|
||
|
|
RUN python -m pip install nvidia-pyindex
|
||
|
|
ADD requirements.txt .
|
||
|
|
RUN python -m pip install -r requirements.txt
|
||
|
|
|
||
|
|
ENV CUDNN_PATH=/usr/local/lib/python3.10/dist-packages/nvidia/cudnn
|
||
|
|
ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.10/dist-packages/nvidia/cudnn/lib
|
||
|
|
|
||
|
|
WORKDIR /app
|
||
|
|
|
||
|
|
ADD go.mod .
|
||
|
|
ADD go.sum .
|
||
|
|
ADD main.go .
|
||
|
|
ADD logic logic
|
||
|
|
|
||
|
|
RUN go install || true
|
||
|
|
|
||
|
|
CMD ["go", "run", "."]
|