mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Use Magma-cuda 12.8 for libtorch (#146019)
https://github.com/pytorch/pytorch/issues/145570 Build failure for libtorch wheel `CUDAContext.cpp:(.text+0x157): additional relocation overflows omitted from the output /usr/bin/ld: failed to convert GOTPCREL relocation; relink with --no-relax collect2: error: ld returned 1 exit status` Unsure if this is related, fixing as a start Pull Request resolved: https://github.com/pytorch/pytorch/pull/146019 Approved by: https://github.com/eqy
This commit is contained in:
parent
2811f33d12
commit
08d88127fe
|
|
@ -7,12 +7,6 @@ function do_install() {
|
|||
cuda_version=$1
|
||||
cuda_version_nodot=${1/./}
|
||||
|
||||
# Temporary WAR to be updated for CUDA 12.8
|
||||
if [ "$cuda_version_nodot" == "128" ]; then
|
||||
# Set it to 12.6 if it matches
|
||||
cuda_version_nodot="126"
|
||||
fi
|
||||
|
||||
MAGMA_VERSION="2.6.1"
|
||||
magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-1.tar.bz2"
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user