ubuntu18+2080ti+cuda+tensorflow+cudnn安装

最新推荐文章于 2022-08-07 15:23:59 发布

克己|

最新推荐文章于 2022-08-07 15:23:59 发布

阅读量297

点赞数 1

本文链接：https://blog.csdn.net/qq_25348431/article/details/110318296

版权

本文详细记录了在Ubuntu系统中配置TensorFlow环境的步骤，包括安装NVIDIA显卡驱动（430版本）、CUDA、CUDNN及指定版本的TensorFlow（1.13.0），旨在避免常见错误，帮助读者顺利搭建环境。

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

Ubuntu下的TensorFlow环境配了好久，踩坑不少。

安装前先去看看TensorFlow和cuda、cudnn的对应版本关系，省的装错了不匹配，还要重新卸载，卸载时候又是问题一大堆。。

这里有对应关系https://blog.csdn.net/qq_34840129/article/details/93913901

1.装显卡驱动，
保守起见，我装的430的（最新的是440）
先删除可能存在的已有驱动，裸的Ubuntu应该是没有驱动的，不过安全起见还是走一下程序

sudo apt-get purge nvidia-*
有些人是从软件和更新里安装的，但是我试了试，没装好。还是用了代码行来装。
把驱动源，源列表更新下

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

安装驱动
sudo apt-get install nvidia-driver-430
重启一下
sudo reboot
测试下
nvidia-smi
nvidia-settings
出现版本提示界面和图形界面就是成功了

这一步基本不会出现什么问题。

2.装cuda

https://developer.nvidia.com/cuda-downloads?target_os=Linux 到历史版本里找到自己需要的版本
下载.run文件
sodu sh *******(你现在的.run文件)

前面是个声明，按着enter键，把声明刷一遍就好了。接来下选accept 注意出现Install NVIDIA Accelerated Graphics Driver for *****？它是想要你装驱动，已经装过了选N! 之后的选Y

接下来把环境变量配置一下
sudo vim ~/.bashrc
尾部加上
export CUDA_HOME=/usr/local/cuda
export PATH=P A T H : PATH:PATH:CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存退出
终端输入 source ~/.bashrc

3.安装cudnn
下载对应的cudnn版本 .deb版本的
sudo dpkg -i *****(你的版本)

4.装TensorFlow

先装虚拟环境 sudo pip3 install -U virtualenv -i （加个镜像）
构建虚拟环境 sudo virtualenv --system-site-packages-p python3 ./venv
激活虚拟环境source ./venv/bin/avtivate
装TensorFlow 指定下版本
pip install tensorflow-gpu==1.13.0 -i（加镜像）