美文网首页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