CV:查看本地的电脑显卡是否支持GPU以及需要安装匹配的tensorflow版本、CUDA版本等
目录
查看本地的电脑显卡是否支持GPU以及tensorflow版本
1、第一步,查看自己的电脑显卡是否支持GPU
2、第二步,查看已安装的tensorflow版本,及其所对应的cuDNN和CUDA版本
T1、pip list和conda list
T2、利用python代码查询
3、第三步,安装CUDA
查看本地的电脑显卡是否支持GPU以及tensorflow版本
1、第一步,查看自己的电脑显卡是否支持GPU
2、第二步,查看已安装的tensorflow版本,及其所对应的cuDNN和CUDA版本
T1、pip list和conda list
T2、利用python代码查询
import tensorflow as tf
print(tf.__version__)
tensorflow版本为2.6.0,根据如下所示 ,可知对应cuDNN版本为8.1、CUDA版本为11.2。
https://tensorflow.google.cn/install/source_windows
3、第三步,安装CUDA
CUDA下载官网:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=11



