美文网首页PyTorch阅读笔记
gcc或者g++或者CUDA版本不对应该怎么办?

gcc或者g++或者CUDA版本不对应该怎么办?

作者: CPinging | 来源:发表于2021-01-07 11:49 被阅读0次

注意编译过程请控制版本问题,第一要控制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

相关文章

网友评论

    本文标题:gcc或者g++或者CUDA版本不对应该怎么办?

    本文链接:https://www.haomeiwen.com/subject/zfkgoktx.html