|
5 | 5 | cd /workspace/project/py
|
6 | 6 |
|
7 | 7 | export CXX=g++
|
8 |
| -export CUDA_HOME=/usr/local/cuda-11.6 |
| 8 | +export CUDA_HOME=/usr/local/cuda-11.7 |
9 | 9 | PROJECT_DIR=/workspace/project
|
10 | 10 |
|
11 | 11 | cp -r $CUDA_HOME /usr/local/cuda
|
12 | 12 |
|
13 | 13 | build_py37() {
|
| 14 | + /opt/python/cp37-cp37m/bin/python -m pip install --upgrade pip |
14 | 15 | /opt/python/cp37-cp37m/bin/python -m pip install -r requirements.txt
|
15 | 16 | /opt/python/cp37-cp37m/bin/python setup.py bdist_wheel --release --ci
|
16 |
| - #auditwheel repair --plat manylinux2014_x86_64 |
| 17 | + #LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/python/cp37-cp37m/lib/python3.7/site-packages/torch/lib:/opt/python/cp37-cp37m/lib/python3.7/site-packages/tensorrt/:/usr/local/cuda-11.7/lib64:/usr/local/cuda-11.7/lib64/stubs /opt/python/cp37-cp37m/bin/python -m auditwheel repair --plat manylinux_2_17_x86_64 dist/torch_tensorrt-*-cp37-cp37m-linux_x86_64.whl |
17 | 18 | }
|
18 | 19 |
|
19 | 20 | build_py38() {
|
| 21 | + /opt/python/cp38-cp38/bin/python -m pip install --upgrade pip |
20 | 22 | /opt/python/cp38-cp38/bin/python -m pip install -r requirements.txt
|
21 | 23 | /opt/python/cp38-cp38/bin/python setup.py bdist_wheel --release --ci
|
22 |
| - #auditwheel repair --plat manylinux2014_x86_64 |
| 24 | + #LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/python/cp38-cp38/lib/python3.8/site-packages/torch/lib:/opt/python/cp38-cp38/lib/python3.8/site-packages/tensorrt/:/usr/local/cuda-11.7/lib64:/usr/local/cuda-11.7/lib64/stubs /opt/python/cp38-cp38/bin/python -m auditwheel repair --plat manylinux_2_17_x86_64 dist/torch_tensorrt-*-cp38-cp38-linux_x86_64.whl |
23 | 25 | }
|
24 | 26 |
|
25 | 27 | build_py39() {
|
| 28 | + /opt/python/cp39-cp39/bin/python -m pip install --upgrade pip |
26 | 29 | /opt/python/cp39-cp39/bin/python -m pip install -r requirements.txt
|
27 | 30 | /opt/python/cp39-cp39/bin/python setup.py bdist_wheel --release --ci
|
28 |
| - #auditwheel repair --plat manylinux2014_x86_64 |
| 31 | + #LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/python/cp39-cp39/lib/python3.8/site-packages/torch/lib:/opt/python/cp39-cp39/lib/python3.9/site-packages/tensorrt/:/usr/local/cuda-11.7/lib64:/usr/local/cuda-11.7/lib64/stubs /opt/python/cp39-cp39/bin/python -m auditwheel repair --plat manylinux_2_17_x86_64 dist/torch_tensorrt-*-cp39-cp39-linux_x86_64.whl |
29 | 32 | }
|
30 | 33 |
|
31 | 34 | build_py310() {
|
| 35 | + /opt/python/cp310-cp310/bin/python -m pip install --upgrade pip |
32 | 36 | /opt/python/cp310-cp310/bin/python -m pip install -r requirements.txt
|
33 | 37 | /opt/python/cp310-cp310/bin/python setup.py bdist_wheel --release --ci
|
34 |
| - #auditwheel repair --plat manylinux2014_x86_64 |
| 38 | + #LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/python/cp310-cp310/lib/python3.8/site-packages/torch/lib:/opt/python/cp310-cp310/lib/python3.8/site-packages/tensorrt/:/usr/local/cuda-11.7/lib64:/usr/local/cuda-11.7/lib64/stubs /opt/python/cp38-cp38/bin/python -m auditwheel repair --plat manylinux_2_17_x86_64 dist/torch_tensorrt-*-cp310-cp310-linux_x86_64.whl |
35 | 39 | }
|
36 | 40 |
|
37 | 41 | #build_py311() {
|
38 | 42 | # /opt/python/cp311-cp311/bin/python -m pip install -r requirements.txt
|
39 | 43 | # /opt/python/cp311-cp311/bin/python setup.py bdist_wheel --release --ci
|
40 |
| -# #auditwheel repair --plat manylinux2014_x86_64 |
| 44 | +# #auditwheel repair --plat manylinux_2_17_x86_64 |
41 | 45 | #}
|
42 | 46 |
|
43 | 47 | build_libtorchtrt() {
|
|
0 commit comments