Skip to content

Commit 6649435

Browse files
ptrblckptrblck
authored
remove CUDA 10.2-11.5 builds (#1194)
* remove CUDA 10.2-11.5 builds * remove 11.5 and 11.3 builds
1 parent cd2573d commit 6649435

17 files changed

+13
-485
lines changed

‎.github/workflows/build-conda-images.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ jobs:
2626
runs-on: linux.2xlarge
2727
strategy:
2828
matrix:
29-
cuda_version: ["10.2", "11.3", "11.5", "11.6", "11.7", "11.8", "cpu"]
29+
cuda_version: ["11.6", "11.7", "11.8", "cpu"]
3030
env:
3131
CUDA_VERSION: ${{ matrix.cuda_version }}
3232
steps:

‎.github/workflows/build-libtorch-images.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ jobs:
2828
runs-on: ubuntu-18.04
2929
strategy:
3030
matrix:
31-
cuda_version: ["11.7", "11.6", "11.5", "11.3", "10.2"]
31+
cuda_version: ["11.7", "11.6"]
3232
env:
3333
GPU_ARCH_TYPE: cuda
3434
GPU_ARCH_VERSION: ${{ matrix.cuda_version }}

‎.github/workflows/build-magma-linux.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ jobs:
3030
runs-on: linux.2xlarge
3131
strategy:
3232
matrix:
33-
cuda_version: ["118", "117", "116", "115"]
33+
cuda_version: ["118", "117", "116"]
3434
steps:
3535
- name: Checkout PyTorch builder
3636
uses: actions/checkout@v2

‎.github/workflows/build-manywheel-images.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ jobs:
2828
runs-on: ubuntu-18.04
2929
strategy:
3030
matrix:
31-
cuda_version: ["11.7", "11.6", "11.3"]
31+
cuda_version: ["11.7", "11.6"]
3232
env:
3333
GPU_ARCH_TYPE: cuda
3434
GPU_ARCH_VERSION: ${{ matrix.cuda_version }}

‎analytics/validate_binaries.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,10 @@
66
PLATFORMS = ["osx-64", "linux-64", "win-64"]
77
PYTHON_VERSIONS = ["3.10", "3.9", "3.8", "3.7"]
88
CUDA_CUDNN_VERSION = [
9-
("11.5", "8.3.2"), ("11.3", "8.2.0"), ("11.1", "8.0.5"), ("10.2", "7.6.5"), ("cpu", None)
9+
("11.7", "8.5.0"), ("cpu", None)
1010
]
1111
CHANNEL = "pytorch-test"
12-
VERSION = "1.11.*"
12+
VERSION = "1.13.*"
1313

1414

1515
def generate_expected_builds(platform: str) -> set:
@@ -22,9 +22,6 @@ def generate_expected_builds(platform: str) -> set:
2222

2323
for cuda_version, cudnn_version in CUDA_CUDNN_VERSION:
2424
if platform == "win-64":
25-
if cuda_version == "10.2":
26-
# win does not support cuda 10.2
27-
continue
2825
cudnn_version = "8"
2926

3027
if cuda_version == "cpu":

‎common/install_cuda.sh

Lines changed: 0 additions & 175 deletions
Original file line numberDiff line numberDiff line change
@@ -2,80 +2,6 @@
22

33
set -ex
44

5-
function install_102 {
6-
echo "Installing CUDA 10.2 and CuDNN"
7-
rm -rf /usr/local/cuda-10.2 /usr/local/cuda
8-
# # install CUDA 10.2 in the same container
9-
wget -q http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
10-
chmod +x cuda_10.2.89_440.33.01_linux.run
11-
./cuda_10.2.89_440.33.01_linux.run --extract=/tmp/cuda
12-
rm -f cuda_10.2.89_440.33.01_linux.run
13-
mv /tmp/cuda/cuda-toolkit /usr/local/cuda-10.2
14-
rm -rf /tmp/cuda
15-
rm -f /usr/local/cuda && ln -s /usr/local/cuda-10.2 /usr/local/cuda
16-
17-
# install CUDA 10.2 CuDNN
18-
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
19-
mkdir tmp_cudnn && cd tmp_cudnn
20-
wget -q http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb -O cudnn-dev.deb
21-
wget -q http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb -O cudnn.deb
22-
ar -x cudnn-dev.deb && tar -xvf data.tar.xz
23-
ar -x cudnn.deb && tar -xvf data.tar.xz
24-
mkdir -p cuda/include && mkdir -p cuda/lib64
25-
cp -a usr/include/x86_64-linux-gnu/cudnn_v7.h cuda/include/cudnn.h
26-
cp -a usr/lib/x86_64-linux-gnu/libcudnn* cuda/lib64
27-
mv cuda/lib64/libcudnn_static_v7.a cuda/lib64/libcudnn_static.a
28-
ln -s libcudnn.so.7 cuda/lib64/libcudnn.so
29-
chmod +x cuda/lib64/*.so
30-
cp -a cuda/include/* /usr/local/cuda/include/
31-
cp -a cuda/lib64/* /usr/local/cuda/lib64/
32-
cd ..
33-
rm -rf tmp_cudnn
34-
ldconfig
35-
}
36-
37-
function install_113 {
38-
echo "Installing CUDA 11.3 and CuDNN 8.3"
39-
rm -rf /usr/local/cuda-11.3 /usr/local/cuda
40-
# install CUDA 11.3.1 in the same container
41-
wget -q https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run
42-
chmod +x cuda_11.3.1_465.19.01_linux.run
43-
./cuda_11.3.1_465.19.01_linux.run --toolkit --silent
44-
rm -f cuda_11.3.1_465.19.01_linux.run
45-
rm -f /usr/local/cuda && ln -s /usr/local/cuda-11.3 /usr/local/cuda
46-
47-
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
48-
mkdir tmp_cudnn && cd tmp_cudnn
49-
wget -q https://developer.download.nvidia.com/compute/redist/cudnn/v8.3.2/local_installers/11.5/cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz -O cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz
50-
tar xf cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz
51-
cp -a cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive/include/* /usr/local/cuda/include/
52-
cp -a cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive/lib/* /usr/local/cuda/lib64/
53-
cd ..
54-
rm -rf tmp_cudnn
55-
ldconfig
56-
}
57-
58-
function install_115 {
59-
echo "Installing CUDA 11.5 and CuDNN 8.3"
60-
rm -rf /usr/local/cuda-11.5 /usr/local/cuda
61-
# install CUDA 11.5.0 in the same container
62-
wget -q https://developer.download.nvidia.com/compute/cuda/11.5.0/local_installers/cuda_11.5.0_495.29.05_linux.run
63-
chmod +x cuda_11.5.0_495.29.05_linux.run
64-
./cuda_11.5.0_495.29.05_linux.run --toolkit --silent
65-
rm -f cuda_11.5.0_495.29.05_linux.run
66-
rm -f /usr/local/cuda && ln -s /usr/local/cuda-11.5 /usr/local/cuda
67-
68-
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
69-
mkdir tmp_cudnn && cd tmp_cudnn
70-
wget -q https://developer.download.nvidia.com/compute/redist/cudnn/v8.3.2/local_installers/11.5/cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz -O cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz
71-
tar xf cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz
72-
cp -a cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive/include/* /usr/local/cuda/include/
73-
cp -a cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive/lib/* /usr/local/cuda/lib64/
74-
cd ..
75-
rm -rf tmp_cudnn
76-
ldconfig
77-
}
78-
795
function install_116 {
806
echo "Installing CUDA 11.6 and CuDNN 8.3"
817
rm -rf /usr/local/cuda-11.6 /usr/local/cuda
@@ -139,101 +65,6 @@ function install_118 {
13965
ldconfig
14066
}
14167

142-
function prune_102 {
143-
echo "Pruning CUDA 10.2 and CuDNN"
144-
#####################################################################################
145-
# CUDA 10.2 prune static libs
146-
#####################################################################################
147-
export NVPRUNE="/usr/local/cuda-10.2/bin/nvprune"
148-
export CUDA_LIB_DIR="/usr/local/cuda-10.2/lib64"
149-
150-
export GENCODE="-gencode arch=compute_35,code=sm_35 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75"
151-
export GENCODE_CUDNN="-gencode arch=compute_35,code=sm_35 -gencode arch=compute_37,code=sm_37 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75"
152-
153-
if [[ -n "$OVERRIDE_GENCODE" ]]; then
154-
export GENCODE=$OVERRIDE_GENCODE
155-
fi
156-
157-
# all CUDA libs except CuDNN and CuBLAS (cudnn and cublas need arch 3.7 included)
158-
ls $CUDA_LIB_DIR/ | grep "\.a" | grep -v "culibos" | grep -v "cudart" | grep -v "cudnn" | grep -v "cublas" | grep -v "metis" \
159-
| xargs -I {} bash -c \
160-
"echo {} && $NVPRUNE $GENCODE $CUDA_LIB_DIR/{} -o $CUDA_LIB_DIR/{}"
161-
162-
# prune CuDNN and CuBLAS
163-
$NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcudnn_static.a -o $CUDA_LIB_DIR/libcudnn_static.a
164-
$NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcublas_static.a -o $CUDA_LIB_DIR/libcublas_static.a
165-
$NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcublasLt_static.a -o $CUDA_LIB_DIR/libcublasLt_static.a
166-
167-
#####################################################################################
168-
# CUDA 10.2 prune visual tools
169-
#####################################################################################
170-
export CUDA_BASE="/usr/local/cuda-10.2/"
171-
rm -rf $CUDA_BASE/libnsight $CUDA_BASE/libnvvp $CUDA_BASE/nsightee_plugins $CUDA_BASE/nsight-compute-2019.5.0 $CUDA_BASE/nsight-systems-2019.5.2
172-
173-
}
174-
175-
function prune_113 {
176-
echo "Pruning CUDA 11.3 and CuDNN"
177-
#####################################################################################
178-
# CUDA 11.3 prune static libs
179-
#####################################################################################
180-
export NVPRUNE="/usr/local/cuda-11.3/bin/nvprune"
181-
export CUDA_LIB_DIR="/usr/local/cuda-11.3/lib64"
182-
183-
export GENCODE="-gencode arch=compute_35,code=sm_35 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86"
184-
export GENCODE_CUDNN="-gencode arch=compute_35,code=sm_35 -gencode arch=compute_37,code=sm_37 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86"
185-
186-
if [[ -n "$OVERRIDE_GENCODE" ]]; then
187-
export GENCODE=$OVERRIDE_GENCODE
188-
fi
189-
190-
# all CUDA libs except CuDNN and CuBLAS (cudnn and cublas need arch 3.7 included)
191-
ls $CUDA_LIB_DIR/ | grep "\.a" | grep -v "culibos" | grep -v "cudart" | grep -v "cudnn" | grep -v "cublas" | grep -v "metis" \
192-
| xargs -I {} bash -c \
193-
"echo {} && $NVPRUNE $GENCODE $CUDA_LIB_DIR/{} -o $CUDA_LIB_DIR/{}"
194-
195-
# prune CuDNN and CuBLAS
196-
$NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcublas_static.a -o $CUDA_LIB_DIR/libcublas_static.a
197-
$NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcublasLt_static.a -o $CUDA_LIB_DIR/libcublasLt_static.a
198-
199-
#####################################################################################
200-
# CUDA 11.3 prune visual tools
201-
#####################################################################################
202-
export CUDA_BASE="/usr/local/cuda-11.3/"
203-
rm -rf $CUDA_BASE/libnvvp $CUDA_BASE/nsightee_plugins $CUDA_BASE/nsight-compute-2021.1.0 $CUDA_BASE/nsight-systems-2021.1.3
204-
}
205-
206-
function prune_115 {
207-
echo "Pruning CUDA 11.5 and CuDNN"
208-
#####################################################################################
209-
# CUDA 11.3 prune static libs
210-
#####################################################################################
211-
export NVPRUNE="/usr/local/cuda-11.5/bin/nvprune"
212-
export CUDA_LIB_DIR="/usr/local/cuda-11.5/lib64"
213-
214-
export GENCODE="-gencode arch=compute_35,code=sm_35 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86"
215-
export GENCODE_CUDNN="-gencode arch=compute_35,code=sm_35 -gencode arch=compute_37,code=sm_37 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86"
216-
217-
if [[ -n "$OVERRIDE_GENCODE" ]]; then
218-
export GENCODE=$OVERRIDE_GENCODE
219-
fi
220-
221-
# all CUDA libs except CuDNN and CuBLAS (cudnn and cublas need arch 3.7 included)
222-
ls $CUDA_LIB_DIR/ | grep "\.a" | grep -v "culibos" | grep -v "cudart" | grep -v "cudnn" | grep -v "cublas" | grep -v "metis" \
223-
| xargs -I {} bash -c \
224-
"echo {} && $NVPRUNE $GENCODE $CUDA_LIB_DIR/{} -o $CUDA_LIB_DIR/{}"
225-
226-
# prune CuDNN and CuBLAS
227-
$NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcublas_static.a -o $CUDA_LIB_DIR/libcublas_static.a
228-
$NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcublasLt_static.a -o $CUDA_LIB_DIR/libcublasLt_static.a
229-
230-
#####################################################################################
231-
# CUDA 11.5 prune visual tools
232-
#####################################################################################
233-
export CUDA_BASE="/usr/local/cuda-11.5/"
234-
rm -rf $CUDA_BASE/libnvvp $CUDA_BASE/nsightee_plugins $CUDA_BASE/nsight-compute-2021.3.0 $CUDA_BASE/nsight-systems-2021.3.3
235-
}
236-
23768
function prune_116 {
23869
echo "Pruning CUDA 11.6 and CuDNN"
23970
#####################################################################################
@@ -331,12 +162,6 @@ function prune_118 {
331162
while test $# -gt 0
332163
do
333164
case "$1" in
334-
10.2) install_102; prune_102
335-
;;
336-
11.3) install_113; prune_113
337-
;;
338-
11.5) install_115; prune_115
339-
;;
340165
11.6) install_116; prune_116
341166
;;
342167
11.7) install_117; prune_117

‎conda/Dockerfile

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -44,18 +44,6 @@ FROM base as cuda
4444
RUN rm -rf /usr/local/cuda-*
4545
ADD ./common/install_cuda.sh install_cuda.sh
4646

47-
FROM cuda as cuda10.2
48-
RUN bash ./install_cuda.sh 10.2
49-
ENV DESIRED_CUDA=10.2
50-
51-
FROM cuda as cuda11.3
52-
RUN bash ./install_cuda.sh 11.3
53-
ENV DESIRED_CUDA=11.3
54-
55-
FROM cuda as cuda11.5
56-
RUN bash ./install_cuda.sh 11.5
57-
ENV DESIRED_CUDA=11.5
58-
5947
FROM cuda as cuda11.6
6048
RUN bash ./install_cuda.sh 11.6
6149
ENV DESIRED_CUDA=11.6
@@ -74,9 +62,6 @@ ADD ./common/install_mnist.sh install_mnist.sh
7462
RUN bash ./install_mnist.sh
7563

7664
FROM base as all_cuda
77-
COPY --from=cuda10.2 /usr/local/cuda-10.2 /usr/local/cuda-10.2
78-
COPY --from=cuda11.3 /usr/local/cuda-11.3 /usr/local/cuda-11.3
79-
COPY --from=cuda11.5 /usr/local/cuda-11.5 /usr/local/cuda-11.5
8065
COPY --from=cuda11.6 /usr/local/cuda-11.6 /usr/local/cuda-11.6
8166
COPY --from=cuda11.7 /usr/local/cuda-11.7 /usr/local/cuda-11.7
8267
COPY --from=cuda11.8 /usr/local/cuda-11.8 /usr/local/cuda-11.8

‎conda/build_all_docker.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,6 @@ set -eou pipefail
44

55
TOPDIR=$(git rev-parse --show-toplevel)
66

7-
for CUDA_VERSION in 11.8 11.7 11.6 11.5 11.3 10.2 cpu; do
7+
for CUDA_VERSION in 11.8 11.7 11.6 cpu; do
88
CUDA_VERSION="${CUDA_VERSION}" conda/build_docker.sh
99
done

‎conda/build_pytorch.sh

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -272,12 +272,6 @@ else
272272
elif [[ "$desired_cuda" == "11.6" ]]; then
273273
export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda >=11.6,<11.7 # [not osx]"
274274
export MAGMA_PACKAGE=" - magma-cuda116 # [not osx and not win]"
275-
elif [[ "$desired_cuda" == "11.3" ]]; then
276-
export CONDA_CUDATOOLKIT_CONSTRAINT=" - cudatoolkit >=11.3,<11.4 # [not osx]"
277-
export MAGMA_PACKAGE=" - magma-cuda113 # [not osx and not win]"
278-
elif [[ "$desired_cuda" == "10.2" ]]; then
279-
export CONDA_CUDATOOLKIT_CONSTRAINT=" - cudatoolkit >=10.2,<10.3 # [not osx]"
280-
export MAGMA_PACKAGE=" - magma-cuda102 # [not osx and not win]"
281275
else
282276
echo "unhandled desired_cuda: $desired_cuda"
283277
exit 1

‎conda/pytorch-nightly/build.sh

Lines changed: 1 addition & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -55,24 +55,7 @@ if [[ -n "$build_with_cuda" ]]; then
5555
export TORCH_CUDA_ARCH_LIST="3.7+PTX;5.0"
5656
export USE_STATIC_CUDNN=1 # links cudnn statically (driven by tools/setup_helpers/cudnn.py)
5757

58-
if [[ $CUDA_VERSION == 10* ]]; then
59-
export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;6.1;7.0;7.5"
60-
DEPS_LIST=(/usr/local/cuda-10.2/extras/CUPTI/lib64/libcupti.so.10.2)
61-
elif [[ $CUDA_VERSION == 11.3* ]]; then
62-
export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;6.1;7.0;7.5;8.0;8.6"
63-
#for cuda 11.3 we use cudnn 8.3.2.44 https://docs.nvidia.com/deeplearning/cudnn/release-notes/rel_8.html
64-
#which does not have single static libcudnn_static.a deliverable to link with
65-
export USE_STATIC_CUDNN=0
66-
#for cuda 11.3 include all dynamic loading libraries
67-
DEPS_LIST=(/usr/local/cuda/lib64/libcudnn*.so.8 /usr/local/cuda-11.3/extras/CUPTI/lib64/libcupti.so.11.3)
68-
elif [[ $CUDA_VERSION == 11.5* ]]; then
69-
export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;6.1;7.0;7.5;8.0;8.6"
70-
#for cuda 11.5 we use cudnn 8.3.2.44 https://docs.nvidia.com/deeplearning/cudnn/release-notes/rel_8.html
71-
#which does not have single static libcudnn_static.a deliverable to link with
72-
export USE_STATIC_CUDNN=0
73-
#for cuda 11.5 include all dynamic loading libraries
74-
DEPS_LIST=(/usr/local/cuda/lib64/libcudnn*.so.8 /usr/local/cuda-11.5/extras/CUPTI/lib64/libcupti.so.11.5)
75-
elif [[ $CUDA_VERSION == 11.6* ]]; then
58+
if [[ $CUDA_VERSION == 11.6* ]]; then
7659
export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;6.1;7.0;7.5;8.0;8.6"
7760
#for cuda 11.5 we use cudnn 8.3.2.44 https://docs.nvidia.com/deeplearning/cudnn/release-notes/rel_8.html
7861
#which does not have single static libcudnn_static.a deliverable to link with

‎libtorch/Dockerfile

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -44,18 +44,6 @@ ADD ./common/install_conda.sh install_conda.sh
4444
RUN bash ./install_conda.sh && rm install_conda.sh
4545
RUN /opt/conda/bin/conda install -y cmake=3.18
4646

47-
FROM cuda as cuda10.2
48-
RUN bash ./install_cuda.sh 10.2
49-
RUN bash ./install_magma.sh 10.2
50-
51-
FROM cuda as cuda11.3
52-
RUN bash ./install_cuda.sh 11.3
53-
RUN bash ./install_magma.sh 11.3
54-
55-
FROM cuda as cuda11.5
56-
RUN bash ./install_cuda.sh 11.5
57-
RUN bash ./install_magma.sh 11.5
58-
5947
FROM cuda as cuda11.6
6048
RUN bash ./install_cuda.sh 11.6
6149
RUN bash ./install_magma.sh 11.6

‎libtorch/build_all_docker.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ set -eou pipefail
44

55
TOPDIR=$(git rev-parse --show-toplevel)
66

7-
for cuda_version in 11.7 11.6 11.5 11.3 10.2; do
7+
for cuda_version in 11.7 11.6; do
88
GPU_ARCH_TYPE=cuda GPU_ARCH_VERSION="${cuda_version}" "${TOPDIR}/libtorch/build_docker.sh"
99
done
1010

0 commit comments

Comments
 (0)