本文详细介绍了如何在Linux系统上安装支持GPU的TensorFlow。首先，需确定匹配的cuDNN和CUDA版本，然后下载并安装CUDA，禁用nouveau驱动，接着卸载旧版驱动，安装CUDA并取消openGL选项，再安装cuDNN，并配置环境变量。最后通过运行deviceQuery测试安装是否成功。
tensorflow是研究和开发深度学习的重要框架，仅使用CPU的tensorflow安装十分便捷，但需要支持GPU的安装还有一些需要注意的地方。
1、根据tensorflow的版本确定依赖的cuDNN和cuda版本。从网页https://tensorflow.google.cn/install/source可以查看，如图所示：
2、根据linux版本下载cuda软件包：https://developer.nvidia.com/cuda-toolkit-archive
3、安装准备
(1)、禁用 nouveau驱动
lsmod | grep nouveau 若无内容输出，则禁用成功，否则reboot重启。
(2)、查看、卸载已有安装：使用nvidia-smi
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
