forked from jiuyuan/InfiniTensor
20 lines
904 B
Plaintext
20 lines
904 B
Plaintext
|
FROM ubuntu:22.04
|
||
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
||
|
|
|
||
|
|
# Install dependencies.
|
||
|
|
RUN apt update && apt-get install -y git make cmake build-essential python-is-python3 python-dev-is-python3 python3-pip libdw-dev openssh-client
|
||
|
|
# Generate ssh key.
|
||
|
|
RUN ssh-keygen -t rsa -f /root/.ssh/id_rsa -q -P ""
|
||
|
|
|
||
|
|
# Update pip and switch to Tsinghua source.
|
||
|
|
RUN python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip && pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
|
||
|
|
|
||
|
|
# TODO: Since SSH clone repo requires adding the SSH key to the GitHub profile,
|
||
|
|
# the process of pulling the project and compiling it has been temporarily commented.
|
||
|
|
|
||
|
|
# Download InfiniTensor.
|
||
|
|
# RUN git clone git@github.com:InfiniTensor/InfiniTensor.git /root/InfiniTensor --branch master --single-branch --recursive
|
||
|
|
|
||
|
|
# Build and Install InfiniTensor
|
||
|
|
# RUN cd /root/InfiniTensor && make install-python
|