纸上得来终觉浅,要知此事需躬行!
网上好多关于如何在WIN10下安装GPU版的TensorFlow的教程,然而写对的都没有几个!包括Tensorflow官网上的教程,也是不对的!
几个大坑:
1、TensorFlow通过conda或者pip安装的话如果不指定版本,默认会装最新的版本。而Tensorflow GPU 1.12.0以上的需要CUDA10.1和对应CUDA版本的CudNN。
2、Anaconda的虚拟环境不要用Anaconda Navigator里面的那个Create按钮创建,不然会缺少很多必要的包。建议在Anaconda Prompt命令行下输入:
conda create --name python=3.6.1
python 3.6.1兼容性最好!
3、如果安装的是Visual Studio 2019 Preview,建议卸载掉自带的Microsoft Visual C++ Distribution 2015-2019,手动安装对应的2015版的X64的MVC。
4、nvidia的显卡驱动要更新到最新版本,然后到nvidia控制面板里看下系统信息中的cuda版本,是否加载的是10.1版本的dll。
5、Anaconda安装过程中要把那个加入到系统环境变量的勾打上!