建议在安装之前先确认下面链接中Python的版本以及其对应的tensorflow、CUDA、cuDNN对应的版本。
初学者可以参考这篇文章:
这里我创建了一个名为 TF ,Python版本为3.8的环境
这里我安装的tensorflow-cpu版本为2.6
如果自己想安装其他的Python版本和对应的tensorflow版本,可以参考文章开头链接
1.为了不和其他环境冲突,我们重新创建一个虚拟环境
2.按照版本对应表中的版本,同样选择2.6版本的
import tensorflow as tf
tf.config.list_physical_devices('GPU')
如果一切设置正确,可以看到下面的输出内容:
4.但是这里我报错了,报错如下:
解决方案如下:在此环境中重新安装 cuDNN 和 cudatoolkit
conda install cudnn -c conda-forge
conda install cudatoolkit -c anaconda
5.再次测试,成功!
因篇幅问题不能全部显示,请点此查看更多更全内容