美文网首页
tensorflow 找不到 cuda, 虽然已经安装了

tensorflow 找不到 cuda, 虽然已经安装了

作者: 孙庚辛 | 来源:发表于2023-10-08 12:37 被阅读0次

    说明 cuda 版本太高, tensorflow 用不了高版本的 cuda.
    安装一下 CUDA 11.8 的版本。
    然后安装 tensorflow 和 tensorrt_libs

    Linux 系统在 ~/.bashrc 文件里修改环境变量, 使得 tensorrt_libs 能够被找到

    export PATH=$PATH:/usr/local/cuda-11.8/bin:/home/qst/.local/bin
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64:/home/qst/.local/lib/python3.10/site-packages/tensorrt_libs
    export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.8
    

    相关文章

      网友评论

          本文标题:tensorflow 找不到 cuda, 虽然已经安装了

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