Error when using rfconv

[codyshen0000]

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 -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS_ -D__CUDA_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 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_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]

Looks like pytorch extension issue