zhanghang1989
commented
3 years ago Looks like pytorch extension issue
Open codyshen0000 opened 3 years ago
zhanghang1989
commented
3 years ago Looks like pytorch extension issue
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.