Skip to content

Commit f52379a

Browse files
authored
Fix gitextensions bug
1 parent 537bd6e commit f52379a

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

packages/torch.Dockerfile

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,6 @@ RUN test -n "$TORCHVISION_VERSION"
1515
# Use mamba to speed up conda installs
1616
RUN conda install -c conda-forge mamba
1717

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-
2318
# Build instructions: https://github.com/pytorch/pytorch#from-source
2419
RUN mamba install astunparse numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses
2520
RUN mamba install -c pytorch magma-cuda${CUDA_MAJOR_VERSION}${CUDA_MINOR_VERSION}
@@ -40,7 +35,7 @@ RUN cd /usr/local/src && \
4035
cd pytorch && \
4136
git checkout tags/v$PACKAGE_VERSION && \
4237
git submodule sync && \
43-
git submodule update --init --recursive --jobs 0 && \
38+
git submodule update --init --recursive --jobs 1 && \
4439
python setup.py bdist_wheel
4540

4641
# Install torch which is required before we can build other torch* packages.
@@ -60,7 +55,7 @@ RUN sudo apt-get update && \
6055
cd audio && \
6156
git checkout tags/v$TORCHAUDIO_VERSION && \
6257
git submodule sync && \
63-
git submodule update --init --recursive --jobs 0 && \
58+
git submodule update --init --recursive --jobs 1 && \
6459
python setup.py bdist_wheel
6560

6661
# Build torchtext
@@ -72,7 +67,7 @@ RUN cd /usr/local/src && \
7267
cd text && \
7368
git checkout tags/v$TORCHTEXT_VERSION && \
7469
git submodule sync && \
75-
git submodule update --init --recursive --jobs 0 && \
70+
git submodule update --init --recursive --jobs 1 && \
7671
python setup.py bdist_wheel
7772

7873
# Build torchvision.

0 commit comments

Comments
 (0)