Skip to content

Commit 5c92884

Browse files
author
Wei Wei
committed
reverse workspace
1 parent 91cfc5d commit 5c92884

File tree

1 file changed

+54
-54
lines changed

1 file changed

+54
-54
lines changed

WORKSPACE

Lines changed: 54 additions & 54 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)