听说tensorflow2.0版本比1.x版本有更多的特点,按照bilibili中的日月光华教程安装好了,记录一下所踩到的坑以及解决方案。
- 首先需要满足Nvidia显卡算力不低于3.0:可以在这里看一下自己的显卡型号:https://developer.nvidia.com/cuda-gpus,本人的显卡是GeForce1050,满足要求。
- 安装Anaconda的对应版本:https://www.anaconda.com/distribution/,(选择最新版本的3.7版本,安装过程可以默认全部打对勾)
- 检查Nidiva显卡驱动版本,需要410.X或者更高:https://jingyan.baidu.com/article/eae078276932761fec5485b7.html
- 检查CUDA版本需要是10.0(推荐10.0.130):https://developer.nvidia.com/cuda-toolkit-archive
- 打开Anaconda Prompt输入:conda install tensorflow-gpu==2.0.0
网络如果连通顺利的话就可以完成安装了,下面说说本人遇到的坑,有几个组件下载过程一直卡住,重复了多次最后只剩一个组件反复下载不成功,于是按照报错提示中给出的URL,复制到迅雷中手动完成了下载。
放置到Anaconda Prompt的当前文件夹,输入:conda install 文件全名,完成了安装。
P.S.末尾安利一下网易云课堂中日月光华老师的课,感觉真的很清晰~~