search

zhanghang1989 / RFConv

Rectified Convolution
https://hangzhang.org/RectifiedConv/RectifiedConv.pdf
Apache License 2.0
44 stars 5 forks source link

Error when using rfconv #4

Open codyshen0000 opened 3 years ago

codyshen0000 commented 3 years ago

from .rfconv import RFConv2d File "/home/lab532/Code/Shen/EDSR/src/model/rfconv/init.py", line 1, in from .nn import RFConv2d File "/home/lab532/Code/Shen/EDSR/src/model/rfconv/nn.py", line 17, in from .function import rectify File "/home/lab532/Code/Shen/EDSR/src/model/rfconv/function.py", line 13, in from . import lib File "/home/lab532/Code/Shen/EDSR/src/model/rfconv/lib/init.py", line 17, in build_directory=cwd, verbose=False) File "/home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 921, in load is_python_module) File "/home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 1109, in _jit_compile with_cuda=with_cuda) File "/home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 1209, in _write_ninja_file_and_build_library error_prefix="Error building extension '{}'".format(name)) File "/home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 1436, in _run_ninja_build raise RuntimeError(message) RuntimeError: Error building extension 'enclib_gpu': [1/3] /usr/bin/nvcc -DTORCH_EXTENSION_NAME=enclib_gpu -DTORCH_API_INCLUDE_EXTENSION_H -isystem /home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/include -isystem /home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/include/torch/csrc/api/include -isystem /home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/include/TH -isystem /home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/include/THC -isystem /home/lab532/anaconda3/envs/lab532/include/python3.6m -D_GLIBCXX_USE_CXX11_ABI=0 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' --expt-extended-lambda -std=c++14 -c /home/lab532/Code/Shen/EDSR/src/model/rfconv/lib/rectify_cuda.cu -o rectify_cuda.cuda.o FAILED: rectify_cuda.cuda.o /usr/bin/nvcc -DTORCH_EXTENSION_NAME=enclib_gpu -DTORCH_API_INCLUDE_EXTENSION_H -isystem /home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/include -isystem /home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/include/torch/csrc/api/include -isystem /home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/include/TH -isystem /home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/include/THC -isystem /home/lab532/anaconda3/envs/lab532/include/python3.6m -D_GLIBCXX_USE_CXX11_ABI=0 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' --expt-extended-lambda -std=c++14 -c /home/lab532/Code/Shen/EDSR/src/model/rfconv/lib/rectify_cuda.cu -o rectify_cuda.cuda.o nvcc fatal : Path to libdevice library not specified [2/3] c++ -MMD -MF operator_cuda.o.d -DTORCH_EXTENSION_NAME=enclib_gpu -DTORCH_API_INCLUDE_EXTENSION_H -isystem /home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/include -isystem /home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/include/torch/csrc/api/include -isystem /home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/include/TH -isystem /home/lab532/anaconda3/envs/lab532/lib/python3.6/site-packages/torch/include/THC -isystem /home/lab532/anaconda3/envs/lab532/include/python3.6m -D_GLIBCXX_USE_CXX11_ABI=0 -fPIC -std=c++14 -c /home/lab532/Code/Shen/EDSR/src/model/rfconv/lib/operator_cuda.cpp -o operator_cuda.o ninja: build stopped: subcommand failed.

zhanghang1989 commented 2 years ago

Looks like pytorch extension issue