Skip to content
Snippets Groups Projects
Commit aa334ba7 authored by Grégoire Kubler's avatar Grégoire Kubler
Browse files

Merge branch 'fix_cuda_wheel_too_big' into 'main'

fix : auditwheel repair re enabled and added --exclude flag on all cuda libraries

Closes #32

See merge request !56
parents aca31cb5 852640af
No related branches found
No related tags found
1 merge request!56fix : auditwheel repair re enabled and added --exclude flag on all cuda libraries
Pipeline #60669 passed
...@@ -43,7 +43,7 @@ release:pip:ubuntu: ...@@ -43,7 +43,7 @@ release:pip:ubuntu:
CUDA_MAJOR_VERSION='12' CUDA_MAJOR_VERSION='12'
CUDA_MINOR_VERSION='4' CUDA_MINOR_VERSION='4'
SEARCH_PATH='/home/ubuntu/builds/$CI_RUNNER_SHORT_TOKEN/$CI_CONCURRENT_ID' SEARCH_PATH='/home/ubuntu/builds/$CI_RUNNER_SHORT_TOKEN/$CI_CONCURRENT_ID'
CIBW_REPAIR_WHEEL_COMMAND='' CIBW_REPAIR_WHEEL_COMMAND='auditwheel --verbose repair {wheel} -w {dest_dir} --exclude libcudart.so.12 --exclude libcudnn.so.9 --exclude libcublas.so.12 --exclude libcublasLt.so.12'
parallel: parallel:
matrix: matrix:
......
...@@ -47,7 +47,7 @@ test-command = "pytest {project}/aidge_backend_cuda/unit_tests" ...@@ -47,7 +47,7 @@ test-command = "pytest {project}/aidge_backend_cuda/unit_tests"
# "cp310-manylinux_x86_64" # "cp310-manylinux_x86_64"
# ] # ]
repair-wheel-command = "" repair-wheel-command = "auditwheel --verbose repair {wheel} -w {dest_dir} --exclude libcudart.so.12 --exclude libcudnn.so.9 --exclude libcublas.so.12 --exclude libcublasLt.so.12"
[tool.cibuildwheel.container-engine] [tool.cibuildwheel.container-engine]
# pass command line options to 'docker run' # pass command line options to 'docker run'
...@@ -81,26 +81,19 @@ before-build = [ ...@@ -81,26 +81,19 @@ before-build = [
"echo '\n\n\n\n yum -y install cuda-toolkit-$CUDA_TOOLKIT_VERSION.$ARCH'", "echo '\n\n\n\n yum -y install cuda-toolkit-$CUDA_TOOLKIT_VERSION.$ARCH'",
"yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/$ARCH/cuda-rhel7.repo", "yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/$ARCH/cuda-rhel7.repo",
"yum clean all", "yum clean all",
"yum -y install cuda-toolkit-$CUDA_TOOLKIT_VERSION.$ARCH", "yum install -y --nogpgcheck cuda-toolkit-$CUDA_TOOLKIT_VERSION.$ARCH",
"yum list available | grep cudnn", "yum list available | grep cudnn",
"yum -y install libcudnn$CUDNN_VERSION-cuda-$CUDA_MAJOR_VERSION.$ARCH", "yum install -y --nogpgcheck libcudnn$CUDNN_VERSION-cuda-$CUDA_MAJOR_VERSION.$ARCH",
"yum -y install libcudnn$CUDNN_VERSION-devel-cuda-$CUDA_MAJOR_VERSION.$ARCH", "yum install -y --nogpgcheck libcudnn$CUDNN_VERSION-devel-cuda-$CUDA_MAJOR_VERSION.$ARCH",
"export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH", "export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH",
"export PATH=/usr/local/cuda/bin:$PATH", "export PATH=/usr/local/cuda/bin:$PATH",
"which nvcc", "which nvcc",
"nvcc --version", "nvcc --version",
"echo '\n\n\n\nInstalling required dependencies for aidge_backend_cuda.\n\n'", "echo '\n\n\n\nInstalling required dependencies for aidge_backend_cuda.\n\n'",
"echo '\n\n\n\nInstalling required dependencies for aidge_backend_cuda.\n\n'",
"bash .gitlab/ci/cibuildwheel_build_deps_before_build_wheel.sh /host/$SEARCH_PATH" "bash .gitlab/ci/cibuildwheel_build_deps_before_build_wheel.sh /host/$SEARCH_PATH"
] ]
before-test= [ before-test= [
"export CUDA_TOOLKIT_VERSION=$CUDA_MAJOR_VERSION-$CUDA_MINOR_VERSION",
"echo '\n\n\n\n yum -y install cuda-toolkit-$CUDA_TOOLKIT_VERSION.$ARCH'",
"yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/$ARCH/cuda-rhel7.repo",
"yum clean all",
"yum -y install cuda-toolkit-$CUDA_TOOLKIT_VERSION.$ARCH",
"yum list available | grep cudnn",
"yum -y install libcudnn$CUDNN_VERSION-cuda-$CUDA_MAJOR_VERSION.$ARCH",
"yum -y install libcudnn$CUDNN_VERSION-devel-cuda-$CUDA_MAJOR_VERSION.$ARCH",
"export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH", "export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH",
"export PATH=/usr/local/cuda/bin:$PATH", "export PATH=/usr/local/cuda/bin:$PATH",
"nvcc --version", "nvcc --version",
......
0.3.0 0.3.2
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment