Skip to content

Commit 202de74

Browse files
committed
Change docker container from condabuilder to nvidia/cuda (#497)
That updates objcopy from 2.32 to 2.35 To workaround `objcopy: unrecognized option '--set-section-alignment'` introduced by pytorch/pytorch#124272
1 parent 29e103d commit 202de74

File tree

2 files changed

+11
-4
lines changed

2 files changed

+11
-4
lines changed

.github/workflows/hqq-dtype.yml

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,12 +17,14 @@ jobs:
1717
script: |
1818
echo "::group::Print machine info"
1919
uname -a
20-
if [ $(uname -s) == Darwin ]; then
21-
sysctl machdep.cpu.brand_string
22-
sysctl machdep.cpu.core_count
23-
fi
2420
echo "::endgroup::"
2521
22+
echo "::group::Install newer objcopy that supports --set-section-alignment"
23+
yum install -y devtoolset-10-binutils
24+
export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH
25+
echo "::endgroup::"
26+
27+
2628
echo "::group::Download checkpoints"
2729
# Install requirements
2830
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121

.github/workflows/pull.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -236,6 +236,11 @@ jobs:
236236
nvidia-smi
237237
echo "::endgroup::"
238238
239+
echo "::group::Install newer objcopy that supports --set-section-alignment"
240+
yum install -y devtoolset-10-binutils
241+
export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH
242+
echo "::endgroup::"
243+
239244
echo "::group::Install required packages"
240245
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121
241246
pip install -r ./requirements.txt

0 commit comments

Comments
 (0)