首页 > 其他 > 详细

TensorFlow环境搭建

时间:2019-03-10 16:36:32      阅读:241      评论:0      收藏:0      [点我收藏+]

1、使用pip安装TensorFlow

 第一步安装pip:

先安装python 官网下载地址https://www.python.org在里面选择适合自己的版本

安装python的过程中pip也会随之安装

Ubuntu下安装pip

sudo apt-get install python-pip python-dev

Mac os下安装pip

sudo easy_install pip

sudo easy_install --upgrade six

第二步:安装TensorFlow

pip install tensorflow  # tensorflow后面加上-gpu安装的是GPU版本,不加是CPU版本
pip install -U tensorflow  #升级方式安装

由于是国外镜像这两种方法可能会导致安装失败,因此可使用国内镜像安装

pip install --index-url https://pypi.douban.com/simple tensorflow  # 豆瓣镜像

pip install --index-url http://mirrors.aliyun.com/pypi/simple/ tensorflow  # 阿里云镜像

2、使用Docker安装 

docker run -it tensorflow/tensorflow:2.0  # CPU版本
nvidia-docker run it tensorflow/tensorflow:2.0-gpu    # GPU版本

3、使用Anaconda安装 

第一步:安装Anaconda

国内下载地址: https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

安装教程可参考: https://www.jianshu.com/p/62f155eb6ac5

第二步:创建自己的虚拟环境(也可参考 )

 
conda update conda   # 更新Anaconda
conda create --name <tf_env>  ==  conda create  -n <tf_env>  # tf_venv是要创建虚拟环境名
conda activate tf_venv 激活虚拟环境
conda env list  # 查看所安装的包

第三步:安装tensorflow(注意要切换到自己的虚拟环境)

conda install tensorflow  # 此处最好使用国内镜像

4、检测tensorflow是否安装成功

# 在python编译环境下引用tensorflow,如果不报错说明安装成功
import tensorflow as tf

 

 

TensorFlow环境搭建

原文:https://www.cnblogs.com/lyh-vip/p/10505697.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!