问题：无法调用GPU跑测试
背景：显卡Titan Xp 系统Ubuntu16.04 环境tensorflow2.0.0 py3.6
解决方案：
①先看tf版本和驱动 cuda cudnn版本是否匹配；
②检查conda环境里是否有cudatoolkit和cudnn包，cuda10.1可对应cudnn7.6.4，记得用conda依次安装
[Tensorflow-gpu2.0.0]Could not load dynamic library ‘libnvinfer.so.6’ and ‘libnvinfer-plugin.so.6’
于 2023-04-17 19:19:49 首次发布
当面临无法使用GPU运行TensorFlow测试时，应检查TensorFlow版本与驱动(cudacudnn)是否兼容，并确保conda环境中已安装正确的cudatoolkit和cudnn包，如cuda10.1对应cudnn7.6.4。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
扫一扫
到【灌水乐园】发言
