TensorFlow是机器学习中使用很多的一个框架,最近开始学习TensorFlow,记录一下安装的过程。
一、TensorFlow的安装
Python的安装此处不再赘述。在装好Python和pip的前提下(笔者Python的版本为3.7),首先“win+R”输入“cmd”打开命令提示符。输入pip install tensorflow
,此时pip开始下载TensorFlow,这一步时间可能较长,请耐心等待。此处可能遇到需更新pip的情况,将弹出的更新命令进行复制粘贴后运行更新pip,再次执行pip install tensorflow
命令直到成功。这便是TensorFlow CPU版本的安装。
TensorFlow GPU版的安装与上述类似。“win+R”输入“cmd”打开命令提示符。输入pip install tensorflow-gpu
,耐心等待安装成功。
二、Cuda和cuDNN的安装
TensorFlow GPU版的安装还需要安装Cuda和cuDNN。此处Cuda和cuDNN的版本极其重要,绝大多数的安装错误都是因为安装的版本不对。还是打开“win+R”输入“cmd”打开命令提示符之后输入python
,进入Python环境后输入import tensorflow
,可以看到报错,原因是缺少cuda,如下图所示。
可以看到,系统提示安装cuda的版本为10.0.到NVIDIA官网https://developer.nvidia.com/cuda-toolkit-archive找到对应版本进行下载。运行下载下来的exe文件,如果嫌麻烦的话一路按默认选项点下一步就可以了。或者选择自定义。
取消勾选GeForce Experience,点下一步。
选择安装路径,建议默认(没有尝试安装到其他路径)。点击下一步进行安装。这一步的安装时间也比较长,耐心等待。
安装好cuda后,再次重复上述“import tensorflow”的操作。发现系统提示应该安装cuDNN,并给出了相应的版本。如图所示即64位7.0版本,到NVIDIA官网https://developer.nvidia.com/rdp/cudnn-archive找到对应版本进行下载。
将下载下来的压缩文件进行解压,发现cuda文件夹下有3个文件夹。
将这三个文件夹拷贝到cuda的安装目录下,比如我按默认路径安装的是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0。重名文件进行替换。
再次重复进行上述“import tensorflow”操作,发现程序没有报错。
三、TensorFlow安装的测试
“win+R”输入“cmd”后输入“python”进入python环境后输入如下代码(或在pycharm等测试也可以)
import tensorflow as tf
hello = tf.constant('hello,tensorf')
sess = tf.Session()
print(sess.run(hello))
进行测试,如果输出b’hello,tensorf’,则安装成功。
下图为命令提示符界面的结果。
pycharm的结果如下图所示。
参考文献
【1】风吹过01.win10环境下安装tensorflow.https://blog.csdn.net/qq_24800377/article/details/82856360
【2】W,Haixin.Win10下Tensorflow(GPU版)安装趟坑实录.https://blog.csdn.net/weixin_39290638/article/details/80045236
【3】_Yucen.TensorFlow测试程序.https://blog.csdn.net/qq_14997473/article/details/80511135