@@ -53,45 +53,45 @@ new_local_repository(
5353# Tarballs and fetched dependencies (default - use in cases when building from precompiled bin and tarballs)
5454#############################################################################################################
5555
56- # http_archive(
57- # name = "libtorch",
58- # build_file = "@//third_party/libtorch:BUILD",
59- # sha256 = "8d9e829ce9478db4f35bdb7943308cf02e8a2f58cf9bb10f742462c1d57bf287",
60- # strip_prefix = "libtorch",
61- # urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-cxx11-abi-shared-with-deps-1.11.0%2Bcu113.zip"],
62- # )
63- #
64- # http_archive(
65- # name = "libtorch_pre_cxx11_abi",
66- # build_file = "@//third_party/libtorch:BUILD",
67- # sha256 = "90159ecce3ff451f3ef3f657493b6c7c96759c3b74bbd70c1695f2ea2f81e1ad",
68- # strip_prefix = "libtorch",
69- # urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.11.0%2Bcu113.zip"],
70- # )
56+ http_archive (
57+ name = "libtorch" ,
58+ build_file = "@//third_party/libtorch:BUILD" ,
59+ sha256 = "8d9e829ce9478db4f35bdb7943308cf02e8a2f58cf9bb10f742462c1d57bf287" ,
60+ strip_prefix = "libtorch" ,
61+ urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-cxx11-abi-shared-with-deps-1.11.0%2Bcu113.zip" ],
62+ )
63+
64+ http_archive (
65+ name = "libtorch_pre_cxx11_abi" ,
66+ build_file = "@//third_party/libtorch:BUILD" ,
67+ sha256 = "90159ecce3ff451f3ef3f657493b6c7c96759c3b74bbd70c1695f2ea2f81e1ad" ,
68+ strip_prefix = "libtorch" ,
69+ urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.11.0%2Bcu113.zip" ],
70+ )
7171
7272# Download these tarballs manually from the NVIDIA website
7373# Either place them in the distdir directory in third_party and use the --distdir flag
7474# or modify the urls to "file:///<PATH TO TARBALL>/<TARBALL NAME>.tar.gz
7575
76- # http_archive(
77- # name = "cudnn",
78- # build_file = "@//third_party/cudnn/archive:BUILD",
79- # sha256 = "0e5d2df890b9967efa6619da421310d97323565a79f05a1a8cb9b7165baad0d7",
80- # strip_prefix = "cuda",
81- # urls = [
82- # "https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.2.4/11.4_20210831/cudnn-11.4-linux-x64-v8.2.4.15.tgz",
83- # ],
84- # )
85- #
86- # http_archive(
87- # name = "tensorrt",
88- # build_file = "@//third_party/tensorrt/archive:BUILD",
89- # sha256 = "826180eaaecdf9a7e76116855b9f1f3400ea9b06e66b06a3f6a0747ba6f863ad",
90- # strip_prefix = "TensorRT-8.2.4.2",
91- # urls = [
92- # "https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/8.2.4/tars/tensorrt-8.2.4.2.linux.x86_64-gnu.cuda-11.4.cudnn8.2.tar.gz",
93- # ],
94- # )
76+ http_archive (
77+ name = "cudnn" ,
78+ build_file = "@//third_party/cudnn/archive:BUILD" ,
79+ sha256 = "0e5d2df890b9967efa6619da421310d97323565a79f05a1a8cb9b7165baad0d7" ,
80+ strip_prefix = "cuda" ,
81+ urls = [
82+ "https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.2.4/11.4_20210831/cudnn-11.4-linux-x64-v8.2.4.15.tgz" ,
83+ ],
84+ )
85+
86+ http_archive (
87+ name = "tensorrt" ,
88+ build_file = "@//third_party/tensorrt/archive:BUILD" ,
89+ sha256 = "826180eaaecdf9a7e76116855b9f1f3400ea9b06e66b06a3f6a0747ba6f863ad" ,
90+ strip_prefix = "TensorRT-8.2.4.2" ,
91+ urls = [
92+ "https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/8.2.4/tars/tensorrt-8.2.4.2.linux.x86_64-gnu.cuda-11.4.cudnn8.2.tar.gz" ,
93+ ],
94+ )
9595
9696####################################################################################
9797# Locally installed dependencies (use in cases of custom dependencies or aarch64)
@@ -104,29 +104,29 @@ new_local_repository(
104104# x86_64 python distribution. If using NVIDIA's version just point to the root of the package
105105# for both versions here and do not use --config=pre-cxx11-abi
106106
107- new_local_repository (
108- name = "libtorch" ,
109- path = "/opt/circleci/.pyenv/versions/3.9.4/ lib/python3.9/site -packages/torch" ,
110- build_file = "third_party/libtorch/BUILD"
111- )
107+ # new_local_repository(
108+ # name = "libtorch",
109+ # path = "/usr/local/ lib/python3.6/dist -packages/torch",
110+ # build_file = "third_party/libtorch/BUILD"
111+ # )
112112
113- new_local_repository (
114- name = "libtorch_pre_cxx11_abi" ,
115- path = "/opt/circleci/.pyenv/versions/3.9.4/ lib/python3.9/site -packages/torch" ,
116- build_file = "third_party/libtorch/BUILD"
117- )
113+ # new_local_repository(
114+ # name = "libtorch_pre_cxx11_abi",
115+ # path = "/usr/local/ lib/python3.6/dist -packages/torch",
116+ # build_file = "third_party/libtorch/BUILD"
117+ # )
118118
119- new_local_repository (
120- name = "cudnn" ,
121- path = "/usr/" ,
122- build_file = "@//third_party/cudnn/local:BUILD"
123- )
119+ # new_local_repository(
120+ # name = "cudnn",
121+ # path = "/usr/",
122+ # build_file = "@//third_party/cudnn/local:BUILD"
123+ # )
124124
125- new_local_repository (
126- name = "tensorrt" ,
127- path = "/usr/" ,
128- build_file = "@//third_party/tensorrt/local:BUILD"
129- )
125+ # new_local_repository(
126+ # name = "tensorrt",
127+ # path = "/usr/",
128+ # build_file = "@//third_party/tensorrt/local:BUILD"
129+ # )
130130
131131# #########################################################################
132132# # Testing Dependencies (optional - comment out on aarch64)
0 commit comments