Ubuntu下CUDA,CUDNN和Tensorflow配置

发布网友发布时间：2024-09-27 14:29

共1个回答

热心网友时间：2024-10-04 18:43

这篇文章详细介绍了在Ubuntu环境下配置CUDA、CUDNN和Tensorflow-GPU的具体步骤。首先，确保Ubuntu语言设置为英文，以便于在命令行界面操作。安装过程主要包括删除旧驱动、禁用nouveau驱动、安装NVIDIA驱动、CUDA 9.0和CUDNN，以及Tensorflow-GPU的安装和验证。

以下是关键步骤：

安装NVIDIA驱动：删除旧驱动，禁用nouveau驱动，创建配置文件，更新并重启系统，然后执行相关命令安装驱动并测试。
安装CUDA 9.0：从官网下载并安装，注意选择不安装驱动，重启电脑并配置环境变量。
CUDNN安装：下载对应版本的tgz包，解压后将include和lib文件移动到CUDA目录，创建软链接，验证安装。
Tensorflow-GPU：安装时选择对应版本，设置安装路径，配置环境变量，验证安装。

最后，如果需要卸载CUDA和Tensorflow，只需简单步骤即可。对于非root用户，需要使用用户路径而非系统路径进行安装。建议使用国内镜像源加速Anaconda的安装。