美文网首页
nvidia-apex cuda 版本安装

nvidia-apex cuda 版本安装

作者: ICEFLY_299f | 来源:发表于2021-04-26 10:06 被阅读0次

    当前apex 版本:0.1
    需要环境:

    • gcc-6.5/g++ 6.5
    • 系统cuda版本要与pytorch 的cuda 版本完全一致:
      如果版本pytorch 对应的是cu101,系统里cuda版本是cu100,运行pytorch 没有问题但是安装apex就会报错,更改系统,cuda版本,更改后需要重新安装pytorch

    安装apex

    git clone https://github.com/NVIDIA/apex
    cd ./apex
    pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext"
    

    refer from :

    相关文章

      网友评论

          本文标题:nvidia-apex cuda 版本安装

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