mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
8c798e0622
Summary: Context: https://github.com/pytorch/pytorch/pull/53299#discussion_r587882857 These are the only hand-written parts of this diff: - the addition to `.github/workflows/lint.yml` - the file endings changed in these four files (to appease FB-internal land-blocking lints): - `GLOSSARY.md` - `aten/src/ATen/core/op_registration/README.md` - `scripts/README.md` - `torch/csrc/jit/codegen/fuser/README.md` The rest was generated by running this command (on macOS): ``` git grep -I -l ' $' -- . ':(exclude)**/contrib/**' ':(exclude)third_party' | xargs gsed -i 's/ *$//' ``` I looked over the auto-generated changes and didn't see anything that looked problematic. Pull Request resolved: https://github.com/pytorch/pytorch/pull/53406 Test Plan: This run (after adding the lint but before removing existing trailing spaces) failed: - https://github.com/pytorch/pytorch/runs/2043032377 This run (on the tip of this PR) succeeded: - https://github.com/pytorch/pytorch/runs/2043296348 Reviewed By: walterddr, seemethere Differential Revision: D26856620 Pulled By: samestep fbshipit-source-id: 3f0de7f7c2e4b0f1c089eac9b5085a58dd7e0d97
59 lines
1.9 KiB
Bash
Executable File
59 lines
1.9 KiB
Bash
Executable File
#!/bin/bash
|
|
##############################################################################
|
|
# Example command to build Caffe2 on Tegra X1.
|
|
##############################################################################
|
|
#
|
|
# This script shows how one can build a Caffe2 binary for NVidia's TX1.
|
|
# The build script assumes that you have the most recent libraries installed
|
|
# via the JetPack toolkit available at
|
|
# https://developer.nvidia.com/embedded/jetpack
|
|
# and it assumes that we are starting from a fresh system after the jetpack
|
|
# installation. If you have already installed some of the dependencies, you
|
|
# may be able to skip quite a few of the apt-get installs.
|
|
|
|
CAFFE2_ROOT="$( cd "$(dirname -- "$0")"/.. ; pwd -P)"
|
|
echo "Caffe2 codebase root is: $CAFFE2_ROOT"
|
|
BUILD_ROOT=${BUILD_ROOT:-"$CAFFE2_ROOT/build"}
|
|
mkdir -p $BUILD_ROOT
|
|
echo "Build Caffe2 raspbian into: $BUILD_ROOT"
|
|
|
|
# obtain necessary dependencies
|
|
echo "Installing dependencies."
|
|
sudo apt-get install \
|
|
cmake \
|
|
libgflags-dev \
|
|
libgoogle-glog-dev \
|
|
libprotobuf-dev \
|
|
protobuf-compiler
|
|
|
|
# obtain optional dependencies that are usually useful to have.
|
|
echo "Installing optional dependencies."
|
|
sudo apt-get install \
|
|
libleveldb-dev \
|
|
liblmdb-dev \
|
|
libpython-dev \
|
|
libsnappy-dev \
|
|
python-numpy \
|
|
python-pip \
|
|
python-protobuf
|
|
|
|
# Obtain python hypothesis, which Caffe2 uses for unit testing. Note that
|
|
# the one provided by apt-get is quite old so we install it via pip
|
|
sudo pip install hypothesis
|
|
|
|
# Install the six module, which includes Python 2 and 3 compatibility utilities,
|
|
# and is required for Caffe2
|
|
sudo pip install six
|
|
|
|
# Now, actually build the android target.
|
|
echo "Building caffe2"
|
|
cd $BUILD_ROOT
|
|
|
|
# CUDA_USE_STATIC_CUDA_RUNTIME needs to be set to off so that opencv can be
|
|
# properly used. Otherwise, opencv will complain that opencv_dep_cudart cannot
|
|
# be found.
|
|
cmake "$CAFFE2_ROOT" -DCUDA_USE_STATIC_CUDA_RUNTIME=OFF \
|
|
|| exit 1
|
|
|
|
make -j 4 || exit 1
|