最近我在入门阶段,对比了知乎上和Google上各路大神的说法,觉得相比起pytorch,还是下载tensorflow2版本比较方便,具体信息请看以下链接:
PyTorch vs TensorFlow — spotting the difference – from Google (Editor: Kirill Dubovikov)
Pytorch是否可以替代Tensorflow? - from 知乎
在选择了tensorflow的情况下,我百度了网上的各路教程,具体说法如下:
- Windows上有两种版本的tensorflow,分别是cpu版本的和gpu版本的,二者差距主要在于计算能力不同,如果我没搞错的话,计算能力:cpu版本 < gpu版本,因为我的电脑显卡支持cuda,所以我选择gpu版本。
#如何判断自己的电脑是否支持cuda呢?
#首先在电脑桌面上右键,找到NVIDIA控制面板,点击
#点击左下角的系统信息,会弹出以下窗口,看到黄色标记的驱动程序版本(我的本来是四百三十几,但是不是最新版的,这就造成了我多次下载产生报错,所以这里要重点标注:驱动程序版本一定要更新,具体怎么更新我后面会讲到)
#停留在这个窗口,但是我们点开“组件”,你可以看到黄线标注的dll,那里显示我的电脑支持CUDA10.2及以下版本
#至此,我知道了我的电脑是支持cuda的,那我可以下载tensorflow-gpu版本
2. Windows上安装tenforfl