下载安装tensorflow-gpu及其相关环境并验证安装成功

下载安装tensorflow-gpu及其相关环境并验证安装成功

一.首先需要注意的是自己的电脑是否支持CUDA,以及支持什么版本,才能安装正确的对应的CUDA.

1.桌面右键,选择下图的选项。如果右键没有,点击在左下角的win标志,在所有程序里面找控制面板。

在这里插入图片描述

2.打开控制面板,点击帮助->系统信息,打开下面窗口,点击组件。可以看到自己支持的CUDA版本,我的支持CUDA10。

下载安装tensorflow-gpu及其相关环境并验证安装成功_第1张图片

二.网上有很多安装CUDA的教程,这里不在赘述,下面附上一个介绍的比较详细的博客。注意的是安装CUDA一定要自定义安装路径,否则会找不到安装的地方,我就因为这个问题卸载安装了好几次。

https://blog.csdn.net/qq_37296487/article/details/83028394

三.安装好CUDA,可以准备安装tensorflow了,但是我用cmd命令行安装的tensorflow总是出问题。

下载安装tensorflow-gpu及其相关环境并验证安装成功_第2张图片

所以,使用别人编译的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

下载安装tensorflow-gpu及其相关环境并验证安装成功_第3张图片

下载之后,在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安装成功。

你可能感兴趣的:(python)