CPU下利用pip安装和卸载Tensorflow

卸载TensorFlow

pip uninstall tensorflow

安装指定版本的TensorFlow

pip install tensorflow==1.10

pip安装opencv

pip install opencv-python

ubuntu系统下编译安装TensorFlow

 

ICU的安装

ICU58.1的下载

ICU52.1的下载

sudo mkdir /usr/local/icu
cd icu/source
./configure --prefix=/usr/local/icu
make
sudo make install

配置环境变量:

sudo gedit /etc/profile

在末尾添加

export LD_LIBRARY_PATH=/usr/local/icu/lib:$LD_LIBRARY_PATH

export PATH=/usr/local/icu/bin:$PATH

source /etc/profile

第0步:安装java

ubuntu下配置Java环境

以上配置完java环境变量后,重启一下电脑。

Bazel在编译TensorFlow时,否则会报错。

第1步:安装Bazel

1.1 下载安装Bazel在ubuntu下的安装

1.2 下载下来的bazel.sh文件 是没有执行权限的,所以要改变权限设置

chmod +x bazel-<version>-installer-linux-x86_64.sh

查看一下bazel.sh文件的权限,是否有执行权限:

 1.3 在终端执行以下命令,

./bazel-<version>-installer-linux-x86_64.sh --user

会将bazel安装在 $HOME/bin下,如下图:

1.4 配置 bazel的环境变量,

gedit $HOME/.bashrc

 在文件末尾加入

export PATH=$HOME/bin:$PATH

如下图所示:

1.5 环境变量生效

source $HOME/.bashrc

第2步:下载 NCCL,用来支持 多GPU

下载后解压:

tar -xf nccl_2.3.7-1+cuda10.0_x86_64.txz

 复制到/usr/local/下

sudo cp -R nccl_2.3.7-1+cuda10.0_x86_64 /usr/local

在/usr/local下多了一个 nccl_2.3.7-1+cuda10.0_x86_64的

sudo ldconfig

配置TensorFlow的编译选项

./configure

利用Bazel编译TensorFlow

bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

 

第3步:下载TensorFlow

首先查看一下 命令 git clone的帮助文档,如下:

然后利用 git clone命令下载tensorflow

git clone https://github.com/tensorflow/tensorflow
git clone --recurse-submodules https://github.com/tensorflow/tensorflow

检出r1.12版本 

git checkout r1.12

编译安装TensorFlow的官方文

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐