tensorflow 找不到 cuda, 虽然已经安装了
2023-10-08 本文已影响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
