美文网首页
cuda版本切换

cuda版本切换

作者: 张亦风 | 来源:发表于2019-10-15 15:01 被阅读0次

    在切换cuda版本时

    rm -rf /usr/local/cuda#删除之前创建的软链接
    sudo ln -s /usr/local/cuda-8.0/ /usr/local/cuda
    nvcc --version #查看当前 cuda 版
    安装cudnn
    sudo cp cuda/include/cudnn.h /usr/local/cuda/include/

    sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
    
    sudo chmod a+r /usr/local/cuda/include/cudnn.h
    
    sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
    

    查看cudnn的版本
    cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

    相关文章

      网友评论

          本文标题:cuda版本切换

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