@@ -15,11 +15,6 @@ RUN test -n "$TORCHVISION_VERSION"
15
15
# Use mamba to speed up conda installs
16
16
RUN conda install -c conda-forge mamba
17
17
18
- # Install cudf/cuml so that cudatoolkit upgrades are included in the pytorch build
19
- RUN conda config --add channels nvidia && \
20
- conda config --add channels rapidsai
21
- RUN mamba install -y cudf cuml
22
-
23
18
# Build instructions: https://github.com/pytorch/pytorch#from-source
24
19
RUN mamba install astunparse numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses
25
20
RUN mamba install -c pytorch magma-cuda${CUDA_MAJOR_VERSION}${CUDA_MINOR_VERSION}
@@ -40,7 +35,7 @@ RUN cd /usr/local/src && \
40
35
cd pytorch && \
41
36
git checkout tags/v$PACKAGE_VERSION && \
42
37
git submodule sync && \
43
- git submodule update --init --recursive --jobs 0 && \
38
+ git submodule update --init --recursive --jobs 1 && \
44
39
python setup.py bdist_wheel
45
40
46
41
# Install torch which is required before we can build other torch* packages.
@@ -60,7 +55,7 @@ RUN sudo apt-get update && \
60
55
cd audio && \
61
56
git checkout tags/v$TORCHAUDIO_VERSION && \
62
57
git submodule sync && \
63
- git submodule update --init --recursive --jobs 0 && \
58
+ git submodule update --init --recursive --jobs 1 && \
64
59
python setup.py bdist_wheel
65
60
66
61
# Build torchtext
@@ -72,7 +67,7 @@ RUN cd /usr/local/src && \
72
67
cd text && \
73
68
git checkout tags/v$TORCHTEXT_VERSION && \
74
69
git submodule sync && \
75
- git submodule update --init --recursive --jobs 0 && \
70
+ git submodule update --init --recursive --jobs 1 && \
76
71
python setup.py bdist_wheel
77
72
78
73
# Build torchvision.
0 commit comments