Hi, when I run bash install.sh
There are some problems, I don't know how to solve it, please help me, Thank you.
unning install_lib
running build_ext
building 'correlation_cuda' extension
gcc -pthread -B /media/data2/xly/anaconda3/envs/py36/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include/TH -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include/THC -I/usr/local/cuda/include -I/media/data2/xly/anaconda3/envs/py36/include/python3.6m -c correlation_cuda.cc -o build/temp.linux-x86_64-3.6/correlation_cuda.o -std=c++11 -DTORCH_EXTENSION_NAME=correlation_cuda
cc1plus: warning: command line option ‘-Wstrict-prototypes’ is valid for C/ObjC but not for C++
correlation_cuda.cc:4:35: fatal error: ATen/cuda/CUDAContext.h: No such file or directory
#include <ATen/cuda/CUDAContext.h>
^
compilation terminated.
error: command 'gcc' failed with exit status 1
running install
running bdist_egg
running egg_info
writing resample2d_cuda.egg-info/PKG-INFO
writing dependency_links to resample2d_cuda.egg-info/dependency_links.txt
writing top-level names to resample2d_cuda.egg-info/top_level.txt
reading manifest file 'resample2d_cuda.egg-info/SOURCES.txt'
writing manifest file 'resample2d_cuda.egg-info/SOURCES.txt'
installing library code to build/bdist.linux-x86_64/egg
running install_lib
running build_ext
building 'resample2d_cuda' extension
gcc -pthread -B /media/data2/xly/anaconda3/envs/py36/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include/TH -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include/THC -I/usr/local/cuda/include -I/media/data2/xly/anaconda3/envs/py36/include/python3.6m -c resample2d_cuda.cc -o build/temp.linux-x86_64-3.6/resample2d_cuda.o -std=c++11 -DTORCH_EXTENSION_NAME=resample2d_cuda
cc1plus: warning: command line option ‘-Wstrict-prototypes’ is valid for C/ObjC but not for C++
/usr/local/cuda/bin/nvcc -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include/TH -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include/THC -I/usr/local/cuda/include -I/media/data2/xly/anaconda3/envs/py36/include/python3.6m -c resample2d_kernel.cu -o build/temp.linux-x86_64-3.6/resample2d_kernel.o -gencode arch=compute_50,code=sm_50 -gencode arch=compute_52,code=sm_52 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -DTORCH_EXTENSION_NAME=resample2d_cuda --compiler-options '-fPIC' -std=c++11
resample2d_kernel.cu:3:35: fatal error: ATen/cuda/CUDAContext.h: No such file or directory
#include <ATen/cuda/CUDAContext.h>
^
compilation terminated.
error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1
running install
running bdist_egg
running egg_info
writing channelnorm_cuda.egg-info/PKG-INFO
writing dependency_links to channelnorm_cuda.egg-info/dependency_links.txt
writing top-level names to channelnorm_cuda.egg-info/top_level.txt
reading manifest file 'channelnorm_cuda.egg-info/SOURCES.txt'
writing manifest file 'channelnorm_cuda.egg-info/SOURCES.txt'
installing library code to build/bdist.linux-x86_64/egg
running install_lib
running build_ext
building 'channelnorm_cuda' extension
gcc -pthread -B /media/data2/xly/anaconda3/envs/py36/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include/TH -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include/THC -I/usr/local/cuda/include -I/media/data2/xly/anaconda3/envs/py36/include/python3.6m -c channelnorm_cuda.cc -o build/temp.linux-x86_64-3.6/channelnorm_cuda.o -std=c++11 -DTORCH_EXTENSION_NAME=channelnorm_cuda
cc1plus: warning: command line option ‘-Wstrict-prototypes’ is valid for C/ObjC but not for C++
/usr/local/cuda/bin/nvcc -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include/TH -I/media/data2/xly/anaconda3/envs/py36/lib/python3.6/site-packages/torch/lib/include/THC -I/usr/local/cuda/include -I/media/data2/xly/anaconda3/envs/py36/include/python3.6m -c channelnorm_kernel.cu -o build/temp.linux-x86_64-3.6/channelnorm_kernel.o -gencode arch=compute_52,code=sm_52 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -DTORCH_EXTENSION_NAME=channelnorm_cuda --compiler-options '-fPIC' -std=c++11
channelnorm_kernel.cu:3:35: fatal error: ATen/cuda/CUDAContext.h: No such file or directory
#include <ATen/cuda/CUDAContext.h>
^
compilation terminated.
error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1