注意编译过程请控制版本问题,第一要控制gcc和GNU的版本,不能超过7,设置为7即可:
可以
vim ~/.bashrc
alias gcc='/usr/bin/gcc'
export CC='/usr/bin/gcc-7'
export CXX='/usr/bin/g++-7'
export CUDA_HOME=/usr/local/cuda
export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"
export LIBRARY_PATH="$LIBRARY_PATH:/usr/local/cuda/lib64"
Cuda版本有问题,使用
cmake -D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda
此外,要加prefix在编译的时候,例如python3 setup.py develop --prefix=/home/cping/PyTorch-lib/PYinstall