ninja: build stopped: subcommand failed

[zzw-zwzhang]

Traceback (most recent call last):
File "/home/a/anaconda3/envs/diffusion/lib/python3.8/multiprocessing/process.py", line 315, in _bootstrap
self.run()
File "/home/a/anaconda3/envs/diffusion/lib/python3.8/multiprocessing/process.py", line 108, in run
self._target(*self._args, **self._kwargs)
File "main.py", line 45, in setup
fn(config)
File "main.py", line 74, in main
import run_lib
File "/import/home/a/TSG/CLD-SGM/run_lib.py", line 21, in
from models import ncsnpp
File "/import/home/a/TSG/CLD-SGM/models/ncsnpp.py", line 25, in
from . import utils, layers, layerspp, normalization
File "/import/home/a/TSG/CLD-SGM/models/layerspp.py", line 20, in from . import up_or_down_sampling File "/import/home/a/TSG/CLD-SGM/models/up_or_down_sampling.py", line 25, in from op import upfirdn2d File "/import/home/a/TSG/CLD-SGM/op/init.py", line 9, in from .fused_act import FusedLeakyReLU, fused_leaky_relu File "/import/home/a/TSG/CLD-SGM/op/fused_act.py", line 19, in fused = load( File "/home/a/anaconda3/envs/diffusion/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1079, in load return _jit_compile( File "/home/a/anaconda3/envs/diffusion/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1292, in _jit_compile _write_ninja_file_and_build_library( File "/home/a/anaconda3/envs/diffusion/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1404, in _write_ninja_file_and_build_library _run_ninja_build( File "/home/a/anaconda3/envs/diffusion/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1683, in _run_ninja_build raise RuntimeError(message) from e
RuntimeError: Error building extension 'fused': [1/2] /usr/local/cuda/bin/nvcc -DTORCH_EXTENSION_NAME=fused -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="_gcc" -DPYBIND11_STDLIB="_libstdcpp" -DP YBIND11_BUILD_ABI="_cxxabi1011"

D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_BFLOAT 16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_70,code=compute_70 -gencode=arch=compute_70,code=sm_70 --compiler-options '-fPIC' -std=c++14 -c /import/home/a$ /TSG/CLD-SGM/op/fused_bias_act_kernel.cu -o fused_bias_act_kernel.cuda.o FAILED: fused_bias_act_kernel.cuda.o /usr/local/cuda/bin/nvcc -DTORCH_EXTENSION_NAME=fused -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="_gcc" -DPYBIND11_STDLIB="_libstdcpp" -DPYBIND11_BUILD_ABI="_cxxabi1011" -isystem

a/anaconda3/envs/diffusion/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_70,code=compute_70 -gencode=arch=compute_70,code=sm_70 --compiler-options '-fPIC' -std=c++14 -c /import/home/a/TSG/CLD-SGM/op/fused_bias_act_kernel.cu -o fused_bias_act_kernel.cuda.o In file included from /import/home/a/TSG/CLD-SGM/op/fused_bias_act_kernel.cu:11:0: /home/a/anaconda3/envs/diffusion/lib/python3.8/site-packages/torch/include/ATen/cuda/CUDAContext.h:7:10: fatal error: cublas_v2.h: No such file or directory #include <cublas_v2.h> ^~~~~~~~~~~~~ compilation terminated. ninja: build stopped: subcommand failed.