blob: f35117152c509ff44251ce1b67424800e8f4b018 (
plain)
1
2
3
4
5
6
7
8
9
10
|
build --action_env TF_CUDA_PATHS=/opt/cuda,/usr/lib,/usr
build --action_env CUDA_TOOLKIT_PATH=/opt/cuda
build --action_env NCCL_INSTALL_PATH=/usr
# Small GPUs like Xavier and Tegra: sm_72,sm_87 (ignore).
build --action_env TF_CUDA_COMPUTE_CAPABILITIES=sm_70,sm_75,sm_80,sm_86,sm_89,sm_90,compute_90
build --action_env GCC_HOST_COMPILER_PATH=/usr/bin/gcc-13
build --config=mkl_open_source_only
build --config=cuda
# Build against py3.12.
build --repo_env=HERMETIC_PYTHON_VERSION=3.12
|