experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add preconfig for CUDA 10.1.

Browse Source 
PiperOrigin-RevId: 278806355
Change-Id: I3db1f63a3d7ce1d82a5b2f6926a1950827a8e1f3
This commit is contained in:
Christian Sigg 2019-11-06 01:06:00 -08:00 committed by TensorFlower Gardener
parent e71a724b76
commit f3c0673ad7
3 changed files with 32 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
third_party/toolchains/preconfig/generate/BUILD vendored
View File

@ -115,6 +115,30 @@ tensorflow_rbe_config(
tensorrt_version = "5.1",
)
tensorflow_rbe_config(
name = "ubuntu16.04-py3-gcc7_manylinux2010-cuda10.1-cudnn7-tensorrt6.0",
compiler = "/dt7/usr/bin/gcc",
compiler_prefix = "/usr/bin",
cuda_version = "10.1",
cudnn_version = "7",
os = "ubuntu16.04-manylinux2010",
python_version = "3",
tensorrt_install_path = "/usr",
tensorrt_version = "6.0",
)
tensorflow_rbe_config(
name = "ubuntu16.04-py3-clang_manylinux2010-cuda10.1-cudnn7-tensorrt6.0",
compiler = "/clang_r373795/bin/clang",
cuda_version = "10.1",
cudnn_version = "7",
os = "ubuntu16.04-manylinux2010",
python_version = "3",
sysroot = "/dt7",
tensorrt_install_path = "/usr",
tensorrt_version = "6.0",
)
tensorflow_rbe_config(
name = "ubuntu16.04-py3_opt-gcc5-rocm",
compiler = "gcc",

1
third_party/toolchains/preconfig/generate/containers.bzl vendored
View File

@ -8,5 +8,6 @@ container_digests = {
"cuda10.0-cudnn7-centos6": "sha256:a1909ba09c703340ee0074ce63dd94fe8fea48035a25264677907a609e2375e0",
"cuda10.1-cudnn7-centos6": "sha256:454b899657e87893ee5e68dc0f87df59b6a0a7418ae09cafcc3dd65ac71feca9",
"cuda10.0-cudnn7-ubuntu16.04-manylinux2010": "sha256:fd17f6899ca7c8a978057aabd584ead788f9404a7918ab61f657e59d49e7f5a0",
"cuda10.1-cudnn7-ubuntu16.04-manylinux2010": "sha256:d7ac7a0559d9f11dab9da65ce1ffa07c07acb75d5bfa4000559d17b9b42b4b8a",
"rocm-ubuntu16.04": "sha256:d5cd4120cff3d2a452378aad03746ff5f24699d86cf695c20ee96f366e42975f",
}

7
third_party/toolchains/preconfig/generate/workspace.bzl vendored
View File

@ -60,6 +60,13 @@ def _remote_config_workspace():
digest = container_digests["cuda10.0-cudnn7-ubuntu16.04-manylinux2010"],
)
container_pull(
name = "cuda10.1-cudnn7-ubuntu16.04-manylinux2010",
registry = "gcr.io",
repository = "tensorflow-testing/nosla-cuda10.1-cudnn7-ubuntu16.04-manylinux2010",
digest = container_digests["cuda10.1-cudnn7-ubuntu16.04-manylinux2010"],
)
container_pull(
name = "rocm-ubuntu16.04",
registry = "gcr.io",