机器学习PAI报错怎么解决?

机器学习PAI报错怎么解决?INFO: Found applicable config definition build:dynamic_kernels in file /home/pangjun/BladeDISC_GPU/tf_community/.bazelrc: --define=dynamic_loaded_kernels=true --copt=-DAUTOLOAD_DYNAMIC_KERNELS
Loading:
WARNING: The following configs were expanded more than once: [cuda]. For repeatable flags, repeats are counted twice and may lead to unexpected behavior.
Loading:
Loading: 0 packages loaded
INFO: Build options --action_env, --compilation_mode, --copt, and 2 more have changed, discarding analysis cache.
Analyzing: 2 targets (1 packages loaded, 0 targets configured)
INFO: Analyzed 2 targets (195 packages loaded, 13606 targets configured).
checking cached actions
INFO: Found 1 target and 1 test target...
[0 / 4] [Prepa] BazelWorkspaceStatusAction stable-status.txt
WARNING: /home/pangjun/.cache/bazel/_bazel_pangjun/a92cb0e935d0b101686941713fa06780/external/org_disc_compiler/mlir/disc/BUILD:2133:8: input 'mlir/disc/cutlass' to @org_disc_compiler//mlir/disc:cutlass_header_preprocess is a directory; dependency checking of directories is unsound
[6,483 / 8,118] Compiling llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp; 0s local, remote-cache ... (191 actions, 190 running)
[6,483 / 8,118] Compiling llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp; 1s local, remote-cache ... (192 actions running)
[6,485 / 8,118] Compiling llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp; 2s local, remote-cache ... (191 actions, 190 running)
ERROR: /home/pangjun/.cache/bazel/_bazel_pangjun/a92cb0e935d0b101686941713fa06780/external/org_disc_compiler/mlir/disc/BUILD:2133:8: Executing genrule @org_disc_compiler//mlir/disc:cutlass_header_preprocess failed: (Exit 2): bash failed: error executing command (from target @org_disc_compiler//mlir/disc:cutlass_header_preprocess) /bin/bash -c ... (remaining 1 argument skipped)
cc1plus: fatal error: cuda_runtime.h: No such file or directory 这个需要怎么设置环境变量能让他找到 conda 环境里的cuda runtime？