@@ -232,7 +232,7 @@ commands:
232
232
parameters :
233
233
python-version :
234
234
type : string
235
- default : " 3.9.4 "
235
+ default : " 3.10.9 "
236
236
steps :
237
237
- run :
238
238
name : Set python version
@@ -295,8 +295,7 @@ commands:
295
295
command : |
296
296
export CUDA_HOME=/usr/local/cuda-<< parameters.cuda-version >>/
297
297
mv toolchains/ci_workspaces/WORKSPACE.<< parameters.platform >> WORKSPACE
298
- python3 -m pip install pip==21.3.1
299
- python3 -m pip wheel --verbose --pre . --extra-index-url << parameters.torch-build-index >> -w dist
298
+ python3 -m pip wheel --no-deps --verbose --pre . --extra-index-url << parameters.torch-build-index >> -w dist
300
299
python3 -m pip install dist/torch_tensorrt*
301
300
mkdir -p /tmp/dist/builds
302
301
cp dist/torch_tensorrt* /tmp/dist/builds
@@ -317,8 +316,7 @@ commands:
317
316
command : |
318
317
export CUDA_HOME=/usr/local/cuda-<< parameters.cuda-version >>/
319
318
mv toolchains/ci_workspaces/WORKSPACE.<< parameters.platform >> WORKSPACE
320
- python3 -m pip install pip==21.3.1
321
- python3 -m pip wheel --verbose --pre . --extra-index-url << parameters.torch-build-index >> --config-setting="--build-option=--legacy" -w dist
319
+ python3 -m pip wheel --no-deps --verbose --pre . --extra-index-url << parameters.torch-build-index >> --config-setting="--build-option=--legacy" -w dist
322
320
python3 -m pip install dist/torch_tensorrt*
323
321
mkdir -p /tmp/dist/builds
324
322
cp dist/torch_tensorrt* /tmp/dist/builds
@@ -348,8 +346,7 @@ commands:
348
346
name : Build torch-tensorrt python release package
349
347
command : |
350
348
export CUDA_HOME=/usr/local/cuda-<< parameters.cuda-version >>/
351
- python3 -m pip install pip==21.3.1
352
- python3 -m pip wheel --verbose --pre . --extra-index-url << parameters.torch-build-index >> --config-setting="--build-option=--release" --config-setting="--build-option=--use-cxx11-abi" -w dist
349
+ python3 -m pip wheel --no-deps --verbose --pre . --extra-index-url << parameters.torch-build-index >> --config-setting="--build-option=--release" --config-setting="--build-option=--use-cxx11-abi" -w dist
353
350
python3 -m pip install dist/torch_tensorrt*
354
351
mkdir -p /tmp/dist/builds
355
352
cp dist/torch_tensorrt* /tmp/dist/builds
@@ -360,8 +357,7 @@ commands:
360
357
name : Build torch-tensorrt python package
361
358
command : |
362
359
export CUDA_HOME=/usr/local/cuda-<< parameters.cuda-version >>/
363
- python3 -m pip install pip==21.3.1
364
- python3 -m pip wheel --verbose --pre . --extra-index-url << parameters.torch-build-index >> --config-setting="--build-option=--use-cxx11-abi" -w dist
360
+ python3 -m pip wheel --no-deps --verbose --pre . --extra-index-url << parameters.torch-build-index >> --config-setting="--build-option=--use-cxx11-abi" -w dist
365
361
python3 -m pip install dist/torch_tensorrt*
366
362
mkdir -p /tmp/dist/builds
367
363
cp dist/torch_tensorrt* /tmp/dist/builds
@@ -383,8 +379,7 @@ commands:
383
379
command : |
384
380
export CUDA_HOME=/usr/local/cuda-<< parameters.cuda-version >>/
385
381
mv toolchains/ci_workspaces/WORKSPACE.<< parameters.platform >> WORKSPACE
386
- python3 -m pip install pip==21.3.1
387
- python3 -m pip wheel --verbose --pre . --extra-index-url << parameters.torch-build-index >> --config-setting="--build-option=--fx-only" -w dist
382
+ python3 -m pip wheel --no-deps --verbose --pre . --extra-index-url << parameters.torch-build-index >> --config-setting="--build-option=--fx-only" -w dist
388
383
python3 -m pip install dist/torch_tensorrt*
389
384
mkdir -p /tmp/dist/builds
390
385
cp dist/torch_tensorrt* /tmp/dist/builds
@@ -425,14 +420,14 @@ commands:
425
420
- run :
426
421
name : Build torch-tensorrt release build for NGC
427
422
command : |
428
- docker exec ngc_build_container bash -c "cd /workspace/ && python3 -m pip wheel --verbose --pre . --config-setting="--build-option=--release" --config-setting="--build-option=--use-cxx11-abi" -w dist"
423
+ docker exec ngc_build_container bash -c "cd /workspace/ && python3 -m pip wheel --no-deps -- verbose --pre . --config-setting="--build-option=--release" --config-setting="--build-option=--use-cxx11-abi" -w dist"
429
424
- unless :
430
425
condition : << parameters.release >>
431
426
steps :
432
427
- run :
433
428
name : Build torch-tensorrt debug build for NGC
434
429
command : |
435
- docker exec ngc_build_container bash -c "cd /workspace/py && python3 -m pip wheel --verbose --pre . --config-setting="--build-option=--use-cxx11-abi" -w dist"
430
+ docker exec ngc_build_container bash -c "cd /workspace/py && python3 -m pip wheel --no-deps -- verbose --pre . --config-setting="--build-option=--use-cxx11-abi" -w dist"
436
431
- run :
437
432
name : Collect builds
438
433
command : |
@@ -453,7 +448,7 @@ commands:
453
448
export PATH=$PATH:/usr/local/cuda-11.8/bin
454
449
~/cmake/bin/cmake -S. -Bbuild \
455
450
-DCMAKE_MODULE_PATH=cmake/Module \
456
- -DTorch_DIR=/opt/circleci/.pyenv/versions/3.9.4 /lib/python3.9 /site-packages/torch/share/cmake/Torch \
451
+ -DTorch_DIR=/opt/circleci/.pyenv/versions/3.10.0 /lib/python3.10 /site-packages/torch/share/cmake/Torch \
457
452
-DTensorRT_ROOT=/usr \
458
453
-DCMAKE_BUILD_TYPE=Debug
459
454
cmake --build build -- -j12
@@ -502,7 +497,7 @@ commands:
502
497
name : Run core / C++ tests
503
498
no_output_timeout : 15m
504
499
environment :
505
- LD_LIBRARY_PATH : " /opt/circleci/.pyenv/versions/3.9.4 /lib/python3.9 /site-packages/torch_tensorrt.libs:/home/circleci/project/bazel-project/external/libtorch_pre_cxx11_abi/lib/:/home/circleci/project/bazel-project/external/tensorrt/lib/:/usr/local/cuda-11.8/lib64/:$LD_LIBRARY_PATH"
500
+ LD_LIBRARY_PATH : " /opt/circleci/.pyenv/versions/3.10.9 /lib/python3.10 /site-packages/torch_tensorrt.libs:/home/circleci/project/bazel-project/external/libtorch_pre_cxx11_abi/lib/:/home/circleci/project/bazel-project/external/tensorrt/lib/:/usr/local/cuda-11.8/lib64/:$LD_LIBRARY_PATH"
506
501
command : |
507
502
set -e
508
503
mv toolchains/ci_workspaces/WORKSPACE.<< parameters.platform >> WORKSPACE
@@ -531,7 +526,7 @@ commands:
531
526
name : Run L0 torch-tensorrt python tests
532
527
environment :
533
528
USE_HOST_DEPS : " 1"
534
- PYT_PATH : " /opt/circleci/.pyenv/versions/3.9.4 /lib/python3.9 /site-packages/"
529
+ PYT_PATH : " /opt/circleci/.pyenv/versions/3.10.9 /lib/python3.10 /site-packages/"
535
530
LD_LIBRARY_PATH : " /usr/lib/x86_64-linux-gnu/:/usr/local/cuda-11.8/lib64/:$LD_LIBRARY_PATH"
536
531
command : |
537
532
set -e
@@ -1155,7 +1150,6 @@ jobs:
1155
1150
TRT_VERSION=$(python3 -c "from versions import __tensorrt_version__;print(__tensorrt_version__)")
1156
1151
CUDNN_VERSION=$(python3 -c "from versions import __cudnn_version__;print(__cudnn_version__)")
1157
1152
TORCH_VERSION=$(python3 -c "from torch import __version__;print(__version__.split('+')[0])")
1158
- python3 -m pip install pip==21.3.1
1159
1153
python3 -m pip install -r ~/project/py/requirements.txt
1160
1154
TORCH_VERSION=$(python3 -c "from torch import __version__;print(__version__.split('+')[0])")
1161
1155
mkdir -p ~/project/py/dist/
@@ -1351,7 +1345,7 @@ jobs:
1351
1345
parameters :
1352
1346
python-version :
1353
1347
type : string
1354
- default : " 3.9.4 "
1348
+ default : " 3.10.9 "
1355
1349
1356
1350
cuda-version :
1357
1351
type : string
0 commit comments