gcc或者g++或者CUDA版本不对应该怎么办？
2021-01-07 本文已影响0人
CPinging
注意编译过程请控制版本问题，第一要控制gcc和GNU的版本，不能超过7，设置为7即可：
可以
image.png
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
