Ubuntu18.04安装TensorFlow-gpu1.14

最新推荐文章于 2024-03-21 09:43:51 发布

iefyy

最新推荐文章于 2024-03-21 09:43:51 发布

阅读量1.1w

点赞数 1

本文链接：https://blog.csdn.net/iefyy/article/details/102740388

版权

本文详细介绍了如何在Ubuntu18.04系统中安装NVIDIA驱动430、CUDA10.1、cuDNN7.6，并最终成功安装TensorFlow-GPU1.14。首先禁用secure boot，安装NVIDIA驱动，然后安装CUDA并配置环境变量，接着安装cuDNN并复制相关文件到CUDA目录，最后通过pip安装TensorFlow-GPU1.14并验证安装成功。

摘要由CSDN通过智能技术生成

Ubuntu18.04+NVIDIA430+CUDA10.1+cuDNN7.6+tensorflow-gpu1.14

一.安装英伟达显卡驱动：
查找安装NVIDIA显卡版本为430
1.禁用BIOS中的secure boot；~~禁用nouveau关闭Xserver~~（不需要）
2.输入终端命令：
ubuntu-drivers devices
sudo ubuntu-drivers autoinstall
3.重新启动
查看nvidia显卡版本命令：
nvidia-smi

二.安装对应版本的CUDA
版本cuda10
sudo sh cuda_10.1.243_418.87.00_linux.run
安装cuda时出现existing package manager installation driver found.在安装选项时选择不安装driver即可。
在.bashrc文件添加环境变量命令（export PATH=$PATH：）：
export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH
生效命令：source .bashrc

三.安装CUDA对应版本的CUDNN
下载的版本：cuDNN library for Linu