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.