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

Add new toolchain platforms for CUDA 10.1 Ubuntu 16 RBE.

Browse Source 
PiperOrigin-RevId: 278820018
Change-Id: I15debca03662bc96611fddef00f01f78689fcb79
This commit is contained in:
Christian Sigg 2019-11-06 02:59:28 -08:00 committed by TensorFlower Gardener
parent 8118a5ee8c
commit 7d2f2375a2
1 changed files with 43 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
third_party/toolchains/BUILD vendored
View File

@ -206,6 +206,49 @@ platform(
""" % container_digests["cuda10.0-cudnn7-ubuntu16.04-manylinux2010"],
)
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.1-cudnn7-ubuntu16.04-manylinux2010.
platform(
name = "rbe_cuda10.1-cudnn7-ubuntu16.04-manylinux2010",
constraint_values = [
"@bazel_tools//platforms:x86_64",
"@bazel_tools//platforms:linux",
],
remote_execution_properties = """
properties: {
name: "container-image"
value:"docker://gcr.io/tensorflow-testing/nosla-cuda10.1-cudnn7-ubuntu16.04-manylinux2010@%s"
}
properties: {
name: "Pool"
value: "default"
}
""" % container_digests["cuda10.1-cudnn7-ubuntu16.04-manylinux2010"],
)
# The above platform with GPU support.
platform(
name = "rbe_cuda10.1-cudnn7-ubuntu16.04-manylinux2010-gpu",
constraint_values = [
"@bazel_tools//platforms:x86_64",
"@bazel_tools//platforms:linux",
":gpu_test",
],
remote_execution_properties = """
properties: {
name: "container-image"
value: "docker://gcr.io/tensorflow-testing/nosla-cuda10.1-cudnn7-ubuntu16.04-manylinux2010@%s"
}
properties: {
name: "dockerRuntime"
value: "nvidia"
}
properties: {
name: "Pool"
value: "gpu-pool"
}
""" % container_digests["cuda10.1-cudnn7-ubuntu16.04-manylinux2010"],
)
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.rocm-ubuntu16.04
platform(
name = "rbe_rocm-ubuntu16.04",