@@ -180,13 +180,10 @@ setup_wheel_python() {
180
180
if [[ " $( uname) " == Darwin || " $OSTYPE " == " msys" ]]; then
181
181
eval " $( conda shell.bash hook) "
182
182
conda env remove -n " env$PYTHON_VERSION " || true
183
- if [[ " $PYTHON_VERSION " == 3.9 ]]; then
184
- export CONDA_CHANNEL_FLAGS=" ${CONDA_CHANNEL_FLAGS} -c=conda-forge"
185
- fi
186
183
conda create ${CONDA_CHANNEL_FLAGS} -yn " env$PYTHON_VERSION " python=" $PYTHON_VERSION "
187
184
conda activate " env$PYTHON_VERSION "
188
185
# Install libpng from Anaconda (defaults)
189
- conda install ${CONDA_CHANNEL_FLAGS} -c conda-forge libpng " jpeg<=9b" -y
186
+ conda install ${CONDA_CHANNEL_FLAGS} libpng " jpeg<=9b" -y
190
187
else
191
188
# Install native CentOS libJPEG, LAME, freetype and GnuTLS
192
189
yum install -y libjpeg-turbo-devel lame freetype gnutls
@@ -250,7 +247,7 @@ setup_pip_pytorch_version() {
250
247
# You MUST have populated PYTORCH_VERSION_SUFFIX before hand.
251
248
setup_conda_pytorch_constraint () {
252
249
if [[ -z " $PYTORCH_VERSION " ]]; then
253
- export CONDA_CHANNEL_FLAGS=" -c pytorch-nightly -c pytorch"
250
+ export CONDA_CHANNEL_FLAGS=" ${CONDA_CHANNEL_FLAGS} -c pytorch-nightly -c pytorch"
254
251
export PYTORCH_VERSION=" $( conda search --json ' pytorch[channel=pytorch-nightly]' | \
255
252
python -c " import os, sys, json, re; cuver = os.environ.get('CU_VERSION'); \
256
253
cuver_1 = cuver.replace('cu', 'cuda') if cuver != 'cpu' else cuver; \
@@ -265,7 +262,7 @@ setup_conda_pytorch_constraint() {
265
262
exit 1
266
263
fi
267
264
else
268
- export CONDA_CHANNEL_FLAGS=" -c pytorch -c pytorch-${UPLOAD_CHANNEL} "
265
+ export CONDA_CHANNEL_FLAGS=" ${CONDA_CHANNEL_FLAGS} -c pytorch -c pytorch-${UPLOAD_CHANNEL} "
269
266
fi
270
267
if [[ " $CU_VERSION " == cpu ]]; then
271
268
export CONDA_PYTORCH_BUILD_CONSTRAINT=" - pytorch==$PYTORCH_VERSION ${PYTORCH_VERSION_SUFFIX} "
@@ -277,9 +274,6 @@ setup_conda_pytorch_constraint() {
277
274
if [[ " $OSTYPE " == msys && " $CU_VERSION " == cu92 ]]; then
278
275
export CONDA_CHANNEL_FLAGS=" ${CONDA_CHANNEL_FLAGS} -c defaults -c numba/label/dev"
279
276
fi
280
- if [[ " $PYTHON_VERSION " == 3.9 ]]; then
281
- export CONDA_CHANNEL_FLAGS=" ${CONDA_CHANNEL_FLAGS} -c=conda-forge"
282
- fi
283
277
}
284
278
285
279
# Translate CUDA_VERSION into CUDA_CUDATOOLKIT_CONSTRAINT
0 commit comments