下载安装tensorflow及其相关环境并验证安装成功
一.首先需要注意的是自己的电脑是否支持CUDA,以及支持什么版本,才能安装正确的对应的CUDA.
1.桌面右键,选择下图的选项。如果右键没有,点击在左下角的win标志,在所有程序里面找控制面板。
2.打开控制面板,点击帮助->系统信息,打开下面窗口,点击组件。可以看到自己支持的CUDA版本,我的支持CUDA10。
二.网上有很多安装CUDA的教程,这里不在赘述,下面附上一个介绍的比较详细的博客。注意的是安装CUDA一定要自定义安装路径,否则会找不到安装的地方,我就因为这个问题卸载安装了好几次。
https://blog.csdn.net/qq_37296487/article/details/83028394
三.安装好CUDA,可以准备安装tensorflow了,但是我用cmd命令行安装的tensorflow总是出问题。
所以,使用别人编译的tensorflow来安装,链接如下。
https://github.com/fo40225/tensorflow-windows-wheel
这里面有很多的tensorflow的wheel文件,一定要选择和自己的电脑,CUDA,python等环境相符合的wheel文件下载,否则会运行不了wheel文件,显示wheel文件不支持平台,如下。
tensorflow_gpu-1.12.0-cp36-cp36m-win_amd64.whl is not a supported wheel on this platform.
我安装了python3.7,安装有 Visual Studio code,SIMD是x86_64的,我下载了第一个路径的wheel文件。
如果不清楚自己的电脑是否支持AVX命令集,可以下载CPU-Z软件,附上知乎的一个相关链接。
https://www.zhihu.com/question/56725903
下载之后,在cmd中pip install 你的wheel文件路径/你的wheel文件全名,如下。
pip install D:\soft_ware\tensorflow\tensorflow-1.13.1-cp37-cp37m-win_amd64.whl
成功安装后验证tensorflow是否安装好了。
如果已经配置了python环境变量的不用管,没有配置的在高级系统设置里面的path配置一下python的环境变量,在cmd窗口键入python,依次键入下图代码。
这表明tensorflow安装成功。