|
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