1. 版本选择
tensorflow官网各gpu版本与对应cuda、cudnn版本
可以看到tf2.6推荐cuda11.2版本
2.cuda安装
CUDA Toolkit Archive | NVIDIA Developer
3. cudnn安装
下载对应版本的cudnn
1. 版本选择
tensorflow官网各gpu版本与对应cuda、cudnn版本
可以看到tf2.6推荐cuda11.2版本
2.cuda安装
CUDA Toolkit Archive | NVIDIA Developer
3. cudnn安装
下载对应版本的cudnn