1.简介

由于Tensoflow-gpu对cuda版本严格的要求，TensoFlow-gpu==2.5.0需要CUDA11.2版本才能使用GPU（conda安装的TensorFlow可以不用安装cuDNN，pip方式安装需要）。

1.1测试TensorFlow能否使用GPU

import tensorflow as tf
# 查看gpu和cpu的数量
gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
cpus = tf.config.experimental.list_physical_devices(device_type='CPU')
print(gpus, cpus)

2.安装CUDA11.2

2.1下载安装包

sudo wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run

2.2安装

sudo sh cuda_11.2.0_460.27.04_linux.run

2.3配置环境变量

2.3.1打开bashrc

sudo vim ~/.bashrc

2.3.2末尾加上

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

2.3.3 内容生效

source ~/.bashrc

2.4 测试CUDA11.2是否安装成功

nvcc -V

3.安装cuDNN

3.1官网下载安装包

去官网注册一个账号，下载对应版本的cuDNN。

只需下载cuDNN Library for Linux (x86_64)就行

下载安装包：
cudnn-11.2-linux-x64-v8.1.1.33.tgz

解压得到一个名为cuda的目录

3.2安装cuDNN

cp cuda/lib64/* /usr/local/cuda-11.2/lib64/
cp cuda/include/* /usr/local/cuda-11.2/include/

3.3测试

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

4.完全卸载CUDA和cuDNN

# 到目录下
cd  /usr/local/cuda-11.2/bin
# 执行
sudo ./cuda-uninstaller

继续清楚残余文件

sudo apt-get remove cuda
sudo apt autoremove
sudo apt-get remove cuda*

到文件下

 cd /usr/local/

删掉两个cuda文件夹

大功告成！

Ubuntu20.04安装CUDA和cuDNN相关推荐

Ubuntu16.04 安装 CUDA、CUDNN、OpenCV 并用 Anaconda 配置 Tensorflow 和 Caffe 详细过程（此种方案不好，好的方案是另一篇，基于pyhton2的）
Ubuntu16.04 安装 CUDA.CUDNN.OpenCV 和 Caffe 详细过程(基于Python2,没有anaconda2和3,因为基于anaconda2和3在安装caffe时会报错,很难 ...
ubuntu20.04安装cuda库
ubuntu20.04安装cuda库官网安装参考参考链接:https://blog.csdn.net/h3c4lenovo/article/details/119003405. 参考连接:http ...
Ubuntu16.04 安装 CUDA、CUDNN、OpenCV 并用 Anaconda 配置 Tensorflow 和 Caffe 详细过程
Ubuntu16.04 安装 CUDA.CUDNN.OpenCV 并用 Anaconda 配置 Tensorflow 和 Caffe 详细过程 1.配置环境 Ubuntu16.04 64-bit系统 ...
笔记本显卡1660Ti，Ubuntu18.04安装Cuda、Cudnn、Anaconda、PyTorch、Tensorflow和Keras
/**********************************************************************/ 注:写到最后不能放图片了,安装PyTorch.Tens ...
Ubuntu18.04安装CUDA、cudnn、pycharm、tensorflow-gpu
前言虚拟机里的linux系统似乎不能安装nvidia显卡驱动,在虚拟机Ubuntu系统上跑tensorflow只能使用cpu..看来还得去物理机装双系统或者用服务器了因为,安装nvidia驱动报错 ...
Ubuntu20.04安装CUDA+CUDNN+Conda+PyTorch
步骤: 1.安装显卡驱动: 2.安装CUDA: 3.安装CUDNN: 4.安装Conda: 5.安装Pytorch. 一.系统和硬件信息 1.Ubuntu 20.04 2.显卡:1050Ti 二.安装 ...
kylin focal fossa ubuntu20.04安装 cuda cudnn lib
1 下载风格版ubuntu20写入U盘 2 分区必须建立data backup各10G,其他是/ 3 安装cuda只能装11以上,但是要关闭xserver cuda下载关闭xserver ctrl ...
Ubuntu 16.04 安装 CUDA、CUDNN 和 GPU 版本的 TensorFlow 一般步骤总结
1. 安装显卡驱动安装显卡驱动网上有各种各样的方法,甚至有的还需要更改一些配置文件切换到命令行终端模式进行操作,然而就我多次安装驱动的经验来看,我从来没有这么费劲过.在 Ubuntu 系统中的 Sy ...
避免繁琐步骤，在Ubuntu22.04安装cuda、cudnn及pytorch
1. 换源国内用户建议改为国内源,比如为清华源.阿里源等. 更改后在终端运行: $: sudo apt update 2. 终端运行命令直接安装显卡驱动运行: $: ubuntu-drivers ...

Ubuntu20.04安装CUDA和cuDNN

这里写目录标题