cpp_extention中nvcc命令指定gcc

extra_compile_args["nvcc"] = [
"-ccbin=/usr/bin/gcc-5",
"-DCUDA_HAS_FP16=1",
"-D__CUDA_NO_HALF_OPERATORS__",
"-D__CUDA_NO_HALF_CONVERSIONS__",
"-D__CUDA_NO_HALF2_OPERATORS__",
]

posted on 2020-10-30 14:41  gris  阅读(352)  评论(0)  编辑  收藏  举报

导航