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