-
Notifications
You must be signed in to change notification settings - Fork 247
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge remote-tracking branch 'upstream/dev' into feature/ultra96v2-su…
…pport
- Loading branch information
Showing
278 changed files
with
18,277 additions
and
18,043 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
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
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
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
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 |
---|---|---|
|
@@ -26,10 +26,10 @@ | |
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | ||
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | ||
|
||
FROM pytorch/pytorch:1.7.1-cuda11.0-cudnn8-runtime | ||
FROM ubuntu:jammy-20230126 | ||
LABEL maintainer="Yaman Umuroglu <[email protected]>" | ||
|
||
ARG XRT_DEB_VERSION="xrt_202210.2.13.466_18.04-amd64-xrt" | ||
ARG XRT_DEB_VERSION="xrt_202220.2.14.354_22.04-amd64-xrt" | ||
|
||
WORKDIR /workspace | ||
|
||
|
@@ -46,7 +46,6 @@ RUN apt-get update && \ | |
libsm6 \ | ||
libxext6 \ | ||
libxrender-dev \ | ||
verilator \ | ||
nano \ | ||
zsh \ | ||
rsync \ | ||
|
@@ -58,10 +57,23 @@ RUN apt-get update && \ | |
unzip \ | ||
zip \ | ||
locales \ | ||
lsb-core | ||
lsb-core \ | ||
python3 \ | ||
python-is-python3 \ | ||
python3-pip | ||
RUN echo "StrictHostKeyChecking no" >> /etc/ssh/ssh_config | ||
RUN locale-gen "en_US.UTF-8" | ||
|
||
# install Verilator from source to get the right version | ||
RUN apt-get install -y git perl make autoconf g++ flex bison ccache libgoogle-perftools-dev numactl perl-doc libfl2 libfl-dev zlib1g zlib1g-dev | ||
RUN git clone https://github.com/verilator/verilator | ||
RUN cd verilator && \ | ||
git checkout v4.224 && \ | ||
autoconf && \ | ||
./configure && \ | ||
make -j4 && \ | ||
make install | ||
|
||
# install XRT | ||
RUN wget https://www.xilinx.com/bin/public/openDownload?filename=$XRT_DEB_VERSION.deb -O /tmp/$XRT_DEB_VERSION.deb | ||
RUN apt install -y /tmp/$XRT_DEB_VERSION.deb | ||
|
@@ -72,22 +84,31 @@ RUN rm /tmp/$XRT_DEB_VERSION.deb | |
COPY requirements.txt . | ||
RUN pip install -r requirements.txt | ||
RUN rm requirements.txt | ||
|
||
# install PyTorch | ||
RUN pip install torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116 | ||
|
||
# extra Python package dependencies (for testing and interaction) | ||
RUN pip install pygments==2.4.1 | ||
RUN pip install ipykernel==5.5.5 | ||
RUN pip install jupyter==1.0.0 | ||
RUN pip install pygments==2.14.0 | ||
RUN pip install ipykernel==6.21.2 | ||
RUN pip install jupyter==1.0.0 --ignore-installed | ||
RUN pip install markupsafe==2.0.1 | ||
RUN pip install matplotlib==3.3.1 --ignore-installed | ||
RUN pip install matplotlib==3.7.0 --ignore-installed | ||
RUN pip install pytest-dependency==0.5.1 | ||
RUN pip install sphinx==5.0.2 | ||
RUN pip install sphinx_rtd_theme==0.5.0 | ||
RUN pip install pytest-xdist[setproctitle]==2.4.0 | ||
RUN pip install pytest-parallel==0.1.0 | ||
RUN pip install pytest-xdist[setproctitle]==3.2.0 | ||
RUN pip install pytest-parallel==0.1.1 | ||
RUN pip install "netron>=5.0.0" | ||
RUN pip install pandas==1.1.5 | ||
RUN pip install scikit-learn==0.24.1 | ||
RUN pip install tqdm==4.31.1 | ||
RUN pip install pandas==1.5.3 | ||
RUN pip install scikit-learn==1.2.1 | ||
RUN pip install tqdm==4.64.1 | ||
RUN pip install -e git+https://github.com/fbcotter/[email protected]#egg=dataset_loading | ||
# these versions of pytest and associated plugins allow for stable collection of | ||
# test reports and code coverage reports in HTML | ||
RUN pip install pytest==6.2.5 | ||
RUN pip install pytest-metadata==1.7.0 | ||
RUN pip install pytest-html==3.0.0 | ||
RUN pip install pytest-html-merger==0.0.8 | ||
RUN pip install pytest-cov==4.1.0 | ||
|
||
# extra dependencies from other FINN deps | ||
# installed in Docker image to make entrypoint script go faster | ||
|
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
Oops, something went wrong.