-
Notifications
You must be signed in to change notification settings - Fork 47
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
bf22d79
commit c91eb11
Showing
3 changed files
with
39 additions
and
2 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
../c-cuda/default_props.cfg |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,36 @@ | ||
FROM nvidia/vulkan:1.1.121 | ||
|
||
ENV DEBIAN_FRONTEND noninteractive | ||
ENV PYTHONDONTWRITEBYTECODE 1 | ||
ENV PATH $PATH:/usr/local/nvidia/lib64/bin | ||
ENV LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH | ||
|
||
RUN env > /etc/environment | ||
|
||
RUN rm -f /etc/apt/sources.list.d/* | ||
|
||
RUN apt-get update && apt install -y --no-install-recommends git ca-certificates \ | ||
python3-distutils python3.8-dev vulkan-utils libvulkan-dev g++-8 \ | ||
vim-tiny less netcat-openbsd inetutils-ping curl patch iproute2 \ | ||
g++ libpci3 libnuma-dev make file openssh-server kmod gdb libopenmpi-dev openmpi-bin psmisc \ | ||
autoconf automake autotools-dev libtool \ | ||
zlib1g-dev rename zip unzip librdmacm-dev gnupg \ | ||
&& apt-get clean && rm -rf /var/lib/apt/lists/* | ||
|
||
RUN ln -sf /usr/bin/g++-8 /usr/bin/g++ | ||
RUN ln -sf /usr/bin/gcc-8 /usr/bin/gcc | ||
RUN ln -sf /usr/bin/gcc /usr/bin/x86_64-linux-gnu-gcc | ||
|
||
RUN ln -sf python3.8 /usr/bin/python3 | ||
RUN ln -sf python3 /usr/bin/python | ||
RUN ln -sf python /usr/bin/python.exe | ||
RUN curl -LO https://bootstrap.pypa.io/get-pip.py && python3 get-pip.py && rm -f get-pip.py | ||
|
||
RUN /bin/echo -e "set backspace=indent,eol,start\nset nocompatible\nset ts=4" > /etc/vim/vimrc.tiny | ||
|
||
ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib64:$LD_LIBRARY_PATH | ||
RUN pip3 install --upgrade antares && mkdir -p /root/.local/antares && mv $(antares pwd)/../3rdparty /root/.local/antares/3rdparty && pip3 uninstall antares -y && echo 'exec /antares/main.py "$@"' > /usr/local/bin/antares && chmod a+x /usr/local/bin/antares | ||
|
||
RUN python3 -m pip install cython | ||
RUN python3 -m pip install torch==2.0.0 --index-url https://download.pytorch.org/whl/cpu | ||
RUN python3 -m pip install setuptools |