Wicher_D的博客

原创 tensorflow使用gpu加速

所以，这里需要根据我们前面的python版本确定对应的CUDA和cuDNN，这里我们使用了python3.9，所以确定对应的CUDA为11.2、cuDNN为8.1。这里需要注意的是，tensorflow是我自定义的环境名，可定义为你喜欢的不带空格和中文的其它名字，python版本要安装tensorflow对应的，可参考下面的网址来确定，这里我以python3.9为例。这里需要注意的是，tensorflow库在2.12及以上的版本中，都取消了对Windows平台的GPU支持（，根据你显卡的版本来下载。