TensorFlow-GPU安装

一开始直接使用 pip install 安装但发现不行，因为之前一直用的torch而且装好了cuda和cudnn
应该是版本对应的问题
cuda、cudnn、TensorFlow-GPU三者版本要对应，链接
cuda版本： nvcc -V（如果显示没有这指令应该是环境变量的问题）
cudnn版本：如果是已经使用过cuda的应该是对应的，新装的话应该也会提醒你需要对应的，顺便说下怎么看cudnn版本，可以在cuda的安装的目录下，include的文件夹里面cudnn_version.h 如下就是8.0.5

#ifndef CUDNN_VERSION_H_
#define CUDNN_VERSION_H_

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 5
1
2
3
4
5
6

版本对应好直接pip吧

pip install --user tensorflow-gpu==2.4.0
1

这里注意加上 --user 不加的话安装的时候会出现报错缺少什么什么权限。