TensorFlow环境搭建
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