@@ -12,7 +12,7 @@ bazel_dep(name = "rules_python", version = "0.34.0")
12
12
python = use_extension("@rules_python//python/extensions:python.bzl", "python")
13
13
python.toolchain(
14
14
ignore_root_user_error = True,
15
- python_version = "3.11 ",
15
+ python_version = "3.12 ",
16
16
)
17
17
18
18
bazel_dep(name = "rules_pkg", version = "1.0.1")
@@ -24,6 +24,12 @@ git_override(
24
24
25
25
local_repository = use_repo_rule("@bazel_tools//tools/build_defs/repo:local.bzl", "local_repository")
26
26
27
+ # External dependency for torch_tensorrt if you already have precompiled binaries.
28
+ local_repository(
29
+ name = "torch_tensorrt",
30
+ path = "/usr/local/lib/python3.12/dist-packages/torch_tensorrt/",
31
+ )
32
+
27
33
new_local_repository = use_repo_rule("@bazel_tools//tools/build_defs/repo:local.bzl", "new_local_repository")
28
34
29
35
@@ -50,13 +56,13 @@ http_archive = use_repo_rule("@bazel_tools//tools/build_defs/repo:http.bzl", "ht
50
56
51
57
new_local_repository(
52
58
name = "libtorch",
53
- path = "/usr/local/lib/python3.10 /dist-packages/torch/",
59
+ path = "/usr/local/lib/python3.12 /dist-packages/torch/",
54
60
build_file = "third_party/libtorch/BUILD"
55
61
)
56
62
57
63
new_local_repository(
58
64
name = "libtorch_pre_cxx11_abi",
59
- path = "/usr/local/lib/python3.10 /dist-packages/torch/",
65
+ path = "/usr/local/lib/python3.12 /dist-packages/torch/",
60
66
build_file = "third_party/libtorch/BUILD"
61
67
)
62
68
0 commit comments