写在前面:初学者一枚,环境搭建花了一些功夫,提供一些经验,希望后来者可以少花些时间吧
1. 显卡和cuda问题,Nvidia3060Ti,我尝试过cuda10.1+tf2.3,也是根据tf官网安装的,不过跑不起来,后来看到有人提到,3060Ti必须要用cuda11,遂卸载cuda10.1,装为cuda11.0,原文链接,不过我想用tf,所以就仅参考了文章关于cuda部分Win10+CUDA11.0+RTX 3060ti GPU深度学习主机 pytorch 环境_DC_Calculator的博客-CSDN博客
可行组合:按顺序安装好:(1)显卡驱动527.56 (2)cuda_11.0 (3)cudnn 8.0 (4)pyuthon 3.8 虚拟环境 (5)tensorflow2.4.0 (6)cudatoolkit 11.0.221 (7) cudnn 8.2.1
所以务必要先根据显卡确定可用的cuda版本,然后根据cuda决定其他,安装按照tf官网的推荐来在 Windows 环境中从源代码构建 | TensorFlow
2. 本以为高枕无忧了,scipy,pandas,sklearn,numpy,这几个包的互相依赖关系,即使一直采用conda解析环境安装,也会有问题,这里仅展示我最后可行的组合吧
可行组合:numpy 1.19.5 pandas 1.4.1 python 3.8.15 scipy 1.9.3 scikit-learn 1.1.0
这是跑tf官方例程初学者的 TensorFlow 2.0 教程 | TensorFlow Core
3. 希望大家的环境都快速搭好~