File tree Expand file tree Collapse file tree 6 files changed +6
-76
lines changed Expand file tree Collapse file tree 6 files changed +6
-76
lines changed Load Diff This file was deleted.
Original file line number Diff line number Diff line change @@ -4,6 +4,6 @@ set -eou pipefail
4
4
5
5
TOPDIR=$( git rev-parse --show-toplevel)
6
6
7
- for CUDA_VERSION in 11.7 11.6 11.5 11.3 10.2 cpu cpu-cxx11-abi ; do
7
+ for CUDA_VERSION in 11.7 11.6 11.5 11.3 10.2 cpu; do
8
8
CUDA_VERSION=" ${CUDA_VERSION} " conda/build_docker.sh
9
9
done
Original file line number Diff line number Diff line change @@ -10,18 +10,12 @@ DEVTOOLSET_VERSION="9"
10
10
if [[ ${CUDA_VERSION: 0: 2} == " 10" ]]; then
11
11
DEVTOOLSET_VERSION=" 7"
12
12
fi
13
- CONDA_LINUX_VERSION=${CONDA_LINUX_VERSION:- }
14
13
15
14
case ${CUDA_VERSION} in
16
15
cpu)
17
16
BASE_TARGET=base
18
17
DOCKER_TAG=cpu
19
18
;;
20
- cpu-cxx11-abi)
21
- BASE_TARGET=base
22
- DOCKER_TAG=cpu-cxx11-abi
23
- CONDA_LINUX_VERSION=" cxx11-abi"
24
- ;;
25
19
all)
26
20
BASE_TARGET=all_cuda
27
21
DOCKER_TAG=latest
@@ -32,12 +26,6 @@ case ${CUDA_VERSION} in
32
26
;;
33
27
esac
34
28
35
- if [[ -n ${CONDA_LINUX_VERSION} ]]; then
36
- DOCKERFILE_SUFFIX=_${CONDA_LINUX_VERSION}
37
- else
38
- DOCKERFILE_SUFFIX=' '
39
- fi
40
-
41
29
(
42
30
set -x
43
31
docker build \
46
34
--build-arg " CUDA_VERSION=${CUDA_VERSION} " \
47
35
--build-arg " DEVTOOLSET_VERSION=${DEVTOOLSET_VERSION} " \
48
36
-t " pytorch/conda-builder:${DOCKER_TAG} " \
49
- -f " ${TOPDIR} /conda/Dockerfile${DOCKERFILE_SUFFIX} " \
37
+ -f " ${TOPDIR} /conda/Dockerfile" \
50
38
${TOPDIR}
51
39
)
52
40
Original file line number Diff line number Diff line change 49
49
build_version=" $2 "
50
50
build_number=" $3 "
51
51
fi
52
- if [[ " $desired_cuda " != cpu ]] && [[ " $desired_cuda " != cpu-cxx11-abi ]] ; then
52
+ if [[ " $desired_cuda " != cpu ]]; then
53
53
desired_cuda=" $( echo $desired_cuda | tr -d cuda. ) "
54
54
fi
55
55
echo " Building cuda version $desired_cuda and pytorch version: $build_version build_number: $build_number "
108
108
if [[ " $OSTYPE " == " darwin" * ]]; then
109
109
DEVELOPER_DIR=/Applications/Xcode9.app/Contents/Developer
110
110
fi
111
- if [[ " $desired_cuda " == ' cpu' ]] || [[ " $desired_cuda " == ' cpu-cxx11-abi ' ]] ; then
111
+ if [[ " $desired_cuda " == ' cpu' ]]; then
112
112
cpu_only=1
113
113
else
114
114
# Switch desired_cuda to be M.m to be consistent with other scripts in
Original file line number Diff line number Diff line change @@ -6,15 +6,8 @@ export CMAKE_PREFIX_PATH=$PREFIX
6
6
export TH_BINARY_BUILD=1 # links CPU BLAS libraries thrice in a row (was needed for some MKL static linkage)
7
7
export PYTORCH_BUILD_VERSION=$PKG_VERSION
8
8
export PYTORCH_BUILD_NUMBER=$PKG_BUILDNUM
9
- if [[ " $DESIRED_DEVTOOLSET " == * " cxx11-abi" * ]]; then
10
- export _GLIBCXX_USE_CXX11_ABI=1
11
- export USE_LLVM=" /opt/llvm"
12
- export LLVM_DIR=" $USE_LLVM /lib/cmake/llvm"
13
- else
14
- export _GLIBCXX_USE_CXX11_ABI=0
15
- export USE_LLVM=" /opt/llvm_no_cxx11_abi"
16
- export LLVM_DIR=" $USE_LLVM /lib/cmake/llvm"
17
- fi
9
+ export USE_LLVM=" /opt/llvm_no_cxx11_abi"
10
+ export LLVM_DIR=" $USE_LLVM /lib/cmake/llvm"
18
11
19
12
# set OPENSSL_ROOT_DIR=/opt/openssl if it exists
20
13
if [[ -e /opt/openssl ]]; then
Original file line number Diff line number Diff line change 70
70
- DEVELOPER_DIR
71
71
- DEBUG
72
72
- USE_FBGEMM
73
- - DESIRED_DEVTOOLSET
74
73
- USE_GLOO_WITH_OPENSSL # [unix]
75
74
- USE_SCCACHE # [win]
76
75
- USE_DISTRIBUTED # [unix]
You can’t perform that action at this time.
0 commit comments