TensorFlow安装教程:GPU和CPU版本安装步骤详解
需积分: 10 71 浏览量
更新于2024-09-09
1
收藏 594KB DOCX 举报
TensorFlow安装教程
TensorFlow是一款开源的人工智能机器学习框架,由Google开发,主要用于大规模机器学习和深度学习。下面是TensorFlow安装教程。
环境配置
在安装TensorFlow之前,需要配置环境。这里使用Anaconda作为环境管理器,conda版本为4.5.11,Python版本为3.6.6,VC++版本为2013。
安装方式
安装TensorFlow有两种方式:pip方式和Anaconda方式。这里推荐使用Anaconda方式。
安装Anaconda
首先,需要下载并安装Anaconda。访问Anaconda的官方网站,下载自己电脑对应的版本,直接运行安装包,下一步就行。在安装过程中,需要选择添加环境变量和把Anaconda下的Python设置为默认版本,这里建议都勾选。
检查Anaconda安装
安装完成后,需要检查Anaconda是否成功安装。打开命令行工具,输入`conda --version`,如果显示Anaconda的版本号,那么安装成功。
检测环境
使用`conda info --envs`命令检测目前安装了哪些环境。
安装Python 3.6
使用`conda create --name tensorflow python=3.6`命令安装Python 3.6。按照提示,激活环境。
激活环境
使用`activate tensorflow`命令激活环境。现在,可以看到命令行工具的头部显示的是tensorflow环境。
检查环境
使用`conda info --envs`命令检查名叫tensorflow的环境已经被成功添加。
检查Python版本
使用`python --version`命令检查新环境中的Python版本。
退出环境
使用`deactivate`命令退出当前环境。
切换环境
使用`activate tensorflow`命令切换到TensorFlow环境。
安装TensorFlow
安装VS2015(CUDA 9.0)或VS2013(CUDA 9.0),然后安装CUDA。解压CUDNN,复制文件夹。使用`pip install tensorflow-gpu==1.10.0`命令安装GPU版本的TensorFlow。
测试
使用`python -c "import tensorflow as tf; print(tf.__version__)"`命令测试TensorFlow是否安装成功。
支持CPU和GPU
TensorFlow既可以支持CPU,也可以支持CPU+GPU。前者的环境需求简单,后者需要额外的支持。TensorFlow是基于VC++开发的,所以需要下载安装Visual C++ Redistributable for Visual Studio来获取MSVCP140.DLL的支持。
注意
在安装TensorFlow之前,需要确保已经安装了CUDA和CUDNN。GPU版本的TensorFlow需要安装CUDA和CUDNN。CPU版本的TensorFlow不需要安装CUDA和CUDNN。
点击了解资源详情
点击了解资源详情
点击了解资源详情
2023-07-15 上传
2023-03-16 上传
2023-09-09 上传
2023-03-16 上传
2023-04-27 上传
weixin_43488919
- 粉丝: 0
- 资源: 1