tensorflow2.4.x gpu支持 cuda11.0 cudnn8.0.5 TensorRT7.2.1

最新推荐文章于 2023-03-07 22:19:00 发布

这道题太难啦

最新推荐文章于 2023-03-07 22:19:00 发布

阅读量1k

点赞数

文章标签：深度学习 tensorflow

本文链接：https://blog.csdn.net/ft193139/article/details/113118860

版权

官网给的样例仅支持CUDA 10.1，tensorflow2.4.x需要cuda11.x，故自行参考N卡官网研究安装CUDA11.0组合（cuda11.0 cudnn8.0.5 TensorRT7.2.1），初测可正常运转：

cuda：11.0

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-ubuntu1604.pin

sudo mv cuda-ubuntu1604.pin /etc/apt/preferences.d/cuda-repository-pin-600

wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda-repo-ubuntu1604-11-0-local_11.0.2-450.51.05-1_amd64.deb

sudo dpkg -i cuda-repo-ubuntu1604-11-0-local_11.0.2-450.51.05-1_amd64.deb

sudo apt-key add /var/cuda-repo-ubuntu1604-11-0-local/7fa2af80.pub

sudo apt-get update

sudo apt-get -y install cuda

cudnn：8.0.5

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-ubuntu1604.pin

sudo mv cuda-ubuntu1604.pin /etc/apt/preferences.d/cuda-repository-pin-600

sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub

sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/ /"

sudo apt-get update

apt-get install libcudnn8=8.0.5.39-1+cuda11.0 --no-install-recommends

apt-get install libcudnn8-dev=8.0.5.39-1+cuda11.0 --no-install-recommends

TensorRT：7.2.1 （下方版本号由TensorRT和cuda版本自由组合）

apt-get install -y --no-install-recommends \

libnvinfer7=7.2.1-1+cuda11.0 \

libnvinfer-dev=7.2.1-1+cuda11.0 \

libnvinfer-plugin7=7.2.1-1+cuda11.0

链接：

vi ~/.bashrc

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11/lib64

export PATH=$PATH:/usr/local/cuda-11/bin

export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11

source ~/.bashrc

测试：

import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

关注