mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
51d0da2802
Summary: Fixes #21026. 1. Improve build docs for Windows 2. Change `BUILD_SHARED_LIBS=ON` for Caffe2 local builds 3. Change to out-source builds for LibTorch and Caffe2 (transferred to #21452) Pull Request resolved: https://github.com/pytorch/pytorch/pull/21190 Differential Revision: D15695223 Pulled By: ezyang fbshipit-source-id: 0ad69d7553a40fe627582c8e0dcf655f6f63bfdf
82 lines
1.6 KiB
Batchfile
82 lines
1.6 KiB
Batchfile
:: #############################################################################
|
|
:: Example command to build on Windows.
|
|
:: #############################################################################
|
|
|
|
:: This script shows how one can build a Caffe2 binary for windows.
|
|
|
|
@echo off
|
|
setlocal
|
|
|
|
SET ORIGINAL_DIR=%cd%
|
|
SET CAFFE2_ROOT=%~dp0%..
|
|
|
|
if NOT DEFINED BUILD_BINARY (
|
|
set BUILD_BINARY=OFF
|
|
)
|
|
|
|
if NOT DEFINED BUILD_SHARED_LIBS (
|
|
:: On CI, we test with BUILD_SHARED_LIBS=OFF.
|
|
:: By default, it will be BUILD_SHARED_LIBS=ON.
|
|
if NOT DEFINED BUILD_ENVIRONMENT (
|
|
set BUILD_SHARED_LIBS=OFF
|
|
)
|
|
)
|
|
|
|
IF NOT DEFINED BUILDING_WITH_TORCH_LIBS (
|
|
set BUILDING_WITH_TORCH_LIBS=OFF
|
|
)
|
|
|
|
if NOT DEFINED CAFFE2_STATIC_LINK_CUDA (
|
|
set CAFFE2_STATIC_LINK_CUDA=OFF
|
|
)
|
|
|
|
if NOT DEFINED CMAKE_BUILD_TYPE (
|
|
set CMAKE_BUILD_TYPE=Release
|
|
)
|
|
|
|
if NOT DEFINED ONNX_NAMESPACE (
|
|
set ONNX_NAMESPACE=onnx_c2
|
|
)
|
|
|
|
if NOT DEFINED TORCH_CUDA_ARCH_LIST (
|
|
set TORCH_CUDA_ARCH_LIST=5.0
|
|
)
|
|
|
|
if NOT DEFINED USE_CUDA (
|
|
set USE_CUDA=OFF
|
|
)
|
|
|
|
if NOT DEFINED USE_OBSERVERS (
|
|
set USE_OBSERVERS=OFF
|
|
)
|
|
|
|
if NOT DEFINED MSVC_Z7_OVERRIDE (
|
|
set MSVC_Z7_OVERRIDE=OFF
|
|
)
|
|
|
|
if NOT DEFINED CMAKE_GENERATOR (
|
|
set CMAKE_GENERATOR=Visual Studio 15 2017
|
|
)
|
|
|
|
:: Install pyyaml for Aten codegen
|
|
pip install pyyaml
|
|
|
|
echo CAFFE2_ROOT=%CAFFE2_ROOT%
|
|
echo CMAKE_GENERATOR=%CMAKE_GENERATOR%
|
|
echo CMAKE_BUILD_TYPE=%CMAKE_BUILD_TYPE%
|
|
|
|
:: Set up cmake. We will skip building the test files right now.
|
|
pushd %CAFFE2_ROOT%
|
|
python tools\build_libtorch.py || goto :label_error
|
|
popd
|
|
|
|
echo "Caffe2 built successfully"
|
|
cd %ORIGINAL_DIR%
|
|
endlocal
|
|
exit /b 0
|
|
|
|
:label_error
|
|
echo "Caffe2 building failed"
|
|
cd %ORIGINAL_DIR%
|
|
endlocal
|
|
exit /b 1
|