tensorflow-gpu cuda cudNN tensorRT 安装
tensorflow-gpu 版本对应关系 https://tensorflow.google.cn/install/source#gpu
CUDA 安装
安装文档 https://docs.nvidia.com/cuda/
linux 安装文档 :https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
win 安装文档 : https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
CUDA 各版本下载地址 : https://developer.nvidia.com/cuda-toolkit-archive
cudNN 安装
安装文档 : https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html
cudNN各版本下载地址 :https://developer.nvidia.com/rdp/cudnn-archive
tensorRT 安装
安装文档 https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html
各版本下载地址:https://developer.nvidia.com/tensorrt-download
tensorflow-gpu 安装
阿里源镜像: https://mirrors.aliyun.com/pypi/simple/tensorflow-gpu/
清华源镜像:https://pypi.tuna.tsinghua.edu.cn/simple/tensorflow-gpu/
或者 tensorflow
阿里源镜像: https://mirrors.aliyun.com/pypi/simple/tensorflow/
清华源镜像:https://pypi.tuna.tsinghua.edu.cn/simple/tensorflow/