參考文檔: https://tensorflow.google.cn/install/pip
首先明確,我們采用python3環境。
1. 先確認本機已安裝好python3的環境
python3 --version
pip3 --version
virtualenv --version
如沒有則安裝以下命令安裝:
$ sudo apt-get install python3-pip python3-dev
$ sudo pip3 install -U virtualenv
2. 創建虛擬環境(推薦)
Python虛擬環境用於將包安裝與系統隔離。
# 通過選擇Python解釋器並創建./venv目錄來保存一個新的虛擬環境:
$ virtualenv --system-site-packages -p python3 ./venv
# 使用特定的shell命令激活虛擬環境:
$ source ./venv/bin/activate # sh, bash, ksh, or zsh
# 當virtualenv激活時,你的shell提示符有(venv)前綴
# 在虛擬的環境中安裝包不會影響主機系統的配置。首先升級pip:
(venv) $ pip install --upgrade pip
(venv) $ pip list
# 然后可以退出虛擬環境:
(venv) $ deactivate # don't exit until you're done using TensorFlow
3. 安裝TensorFlow pip包
從PyPI安裝以下一個TensorFlow軟件包:
- tensorflow —Current release for CPU-only (recommended for beginners)
- tensorflow-gpu —Current release with GPU support (Ubuntu and Windows)
- tf-nightly —Nightly build for CPU-only (unstable)
- tf-nightly-gpu —Nightly build with GPU support (unstable, Ubuntu and Windows)
如果是要安裝GPU版本,需要做一些額外的設置:
# 對於Ubuntu 16.04和可能的其他基於Debian的Linux Distros添加NVIDIA包存儲庫,並使用APT安裝CUDA。
# Add NVIDIA package repository
$ sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
$ wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
$ sudo apt install ./cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
$ wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb
$ sudo apt install ./nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb
$ sudo apt update
# Install CUDA and tools. Include optional NCCL 2.x
$ sudo apt install cuda9.0 cuda-cublas-9-0 cuda-cufft-9-0 cuda-curand-9-0 \
cuda-cusolver-9-0 cuda-cusparse-9-0 libcudnn7=7.2.1.38-1+cuda9.0 \
libnccl2=2.2.13-1+cuda9.0 cuda-command-line-tools-9-0
# Optional: Install the TensorRT runtime (must be after CUDA install)
$ sudo apt update
$ sudo apt install libnvinfer4=4.1.2-1+cuda9.0
虛擬環境安裝:
(venv) $ pip install --upgrade tensorflow # 可能時間會比較長,如果要支持GPU,請安裝tensorflow-gpu
# 驗證安裝:
(venv) $ python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"
安裝成功后,就可以開始學習如何使用了。