ubuntu的tensorflow环境配置

最新推荐文章于 2025-07-13 19:01:29 发布

原创最新推荐文章于 2025-07-13 19:01:29 发布 · 684 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ubuntu #tensorflow

tensorflow 专栏收录该内容

3 篇文章

订阅专栏

本文详细介绍了在Ubuntu系统上安装NVIDIA显卡驱动、CUDA10.1及cuDNN的过程，以及如何配置环境变量，最后安装TensorFlow-GPU以实现GPU加速的深度学习环境。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、N卡驱安装

添加nvidia官方ppa

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

sudo ubuntu-drivers devices // 显示可以安装的nvidia驱动
 
sudo apt install nvidia-xxx //安装想要安装的版本

安装好后测试是否安装成功

nvidia-smi

若出现这个即为成功

二、下载安装cuda10.01，cudnn

1）cuda下载地址:https://developer.nvidia.com/cuda-10.1-download-archive-update2

或者命令下载:

wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run

安装cuda

sudo chmod 777  cuda_10.1.243_418.87.00_linux.run
sudo ./cuda_10.1.243_418.87.00_linux.run

再出现的提示中选择continue和accept,直到出现如下画面

咱们吧显卡驱动上的X去掉就好（前面已经装过驱动了）

配置Cuda环境

1、配置到动态链接库

sudo gedit /etc/ld.so.conf.d/cuda.conf

在打开的文件中添加如下语句：

/usr/local/cuda-10.1/lib64

执行

sudo ldconfig

2、配置到环境变量（不同环境下，配置不同环境变量，可以使用多个cuda版本）

sudo gedit ~/.bashrc

打开文件后在文件末尾添加路径，也就是安装目录：

export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH

然后重新加载.bashrc文件

source ~/.bashrc

2）安装cudnn

cudnn下载链接:https://developer.nvidia.com/rdp/cudnn-download

下载后安装cudnn

安装 cudnn：

tar -zxvf cudnn-10.1-linux-x64-v7.6.5.32.tgz
cd cuda
sudo cp lib64/lib* /usr/local/cuda-10.1/lib64/
sudo cp include/cudnn.h /usr/local/cuda-10.1/include/
cd /usr/local/cuda-10.1/lib64/
sudo chmod +r libcudnn.so.7.6.5 # 自己查看.so的版本
sudo ln -sf libcudnn.so.7.6.5 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so

三、安装tensorflow-gpu