github.com/johnnyeven/libtools@v0.0.0-20191126065708-61829c1adf46/third_party/toolchains/preconfig/generate/workspace.bzl (about) 1 load( 2 "@io_bazel_rules_docker//repositories:repositories.bzl", 3 container_repositories = "repositories", 4 ) 5 load( 6 "@io_bazel_rules_docker//container:container.bzl", 7 "container_pull", 8 ) 9 load(":containers.bzl", "container_digests") 10 11 def _remote_config_workspace(): 12 container_repositories() 13 14 container_pull( 15 name = "centos6", 16 registry = "gcr.io", 17 repository = "tensorflow-testing/nosla-centos6", 18 digest = container_digests["centos6"], 19 ) 20 21 container_pull( 22 name = "ubuntu16.04", 23 registry = "gcr.io", 24 repository = "tensorflow-testing/nosla-ubuntu16.04", 25 digest = container_digests["ubuntu16.04"], 26 ) 27 28 container_pull( 29 name = "cuda10.0-cudnn7-ubuntu14.04", 30 registry = "gcr.io", 31 repository = "tensorflow-testing/nosla-cuda10.0-cudnn7-ubuntu14.04", 32 digest = container_digests["cuda10.0-cudnn7-ubuntu14.04"], 33 ) 34 35 container_pull( 36 name = "cuda10.0-cudnn7-centos6", 37 registry = "gcr.io", 38 repository = "tensorflow-testing/nosla-cuda10.0-cudnn7-centos6", 39 digest = container_digests["cuda10.0-cudnn7-centos6"], 40 ) 41 42 container_pull( 43 name = "cuda10.1-cudnn7-centos6", 44 registry = "gcr.io", 45 repository = "tensorflow-testing/nosla-cuda10.1-cudnn7-centos6", 46 digest = container_digests["cuda10.1-cudnn7-centos6"], 47 ) 48 49 container_pull( 50 name = "ubuntu16.04-manylinux2010", 51 registry = "gcr.io", 52 repository = "tensorflow-testing/nosla-ubuntu16.04-manylinux2010", 53 digest = container_digests["ubuntu16.04-manylinux2010"], 54 ) 55 56 container_pull( 57 name = "cuda10.0-cudnn7-ubuntu16.04-manylinux2010", 58 registry = "gcr.io", 59 repository = "tensorflow-testing/nosla-cuda10.0-cudnn7-ubuntu16.04-manylinux2010", 60 digest = container_digests["cuda10.0-cudnn7-ubuntu16.04-manylinux2010"], 61 ) 62 63 container_pull( 64 name = "rocm-ubuntu16.04", 65 registry = "gcr.io", 66 repository = "tensorflow-testing/nosla-rocm-ubuntu16.04", 67 digest = container_digests["rocm-ubuntu16.04"], 68 ) 69 70 remote_config_workspace = _remote_config_workspace