win10 安装 GPU+Tensorflow


安装环境: 惠普暗影精灵4   1050Ti  Win10

cpu的安装极其容易,pip或者conda均可

gpu的安装:

pip安装极其缓慢。采用清华大学的镜像文件,倒是光速下载,但是在引入包的时候遇到ddl缺失的问题,在tensorflow官网上没有找到合适的解决办法,但是问题被列出来了。。详见https://github.com/tensorflow/tensorflow/issues/22794。。

决定改用conda安装。基本思想参考了这位大佬https://www.pugetsystems.com/labs/hpc/The-Best-Way-to-Install-TensorFlow-with-GPU-Support-on-Windows-10-Without-Installing-CUDA-1187/。。conda安装不需要单独安装cuda和cudnn。甚至于这篇博文都有些落伍了。现在几乎一条指令完事。

conda install tensorflow-gpu

会根据你的python版本自动下载相应的tensorflow,cuda和cudnn。

cuda和cudnn无需手动安装,无疑是太方便了。基本原理是通过conda创造了一个虚拟环境,conda自动在则会个环境里帮你解决了包的依赖问题。