1. 程式人生 > >安裝特定版本的TensorFlow(以Linux平臺為例)

安裝特定版本的TensorFlow(以Linux平臺為例)

    在科研學習的過程中,經常出現一些需要使用其他研究人員既有演算法程式碼的情況。這些程式碼很有可能使用版本較早的TensorFlow完成,由於TensorFlow更新很快,所以需要將TensorFlow恢復到較早版本。
    以Linux平臺上的pip安裝為例,在命令列終端輸入以下命令:

# 64位CPU版本
export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.0.0-cp27-none-linux_x86_64.whl

# 64位GPU版本
export TF_BINARY_URL=https://storage.googleapis
.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.0-cp27-none-linux_x86_64.whl

    上邊的指令指定了程式包下載的路徑。通過改動“/tensorflow-**-cp-none-linux_x86_64.whl”中的版本號——星號代指版本編號,我們可以下載其他版本的TensorFlow。如,tensorflow-1.0.0, tensorflow-1.1.1等等。
    指定路徑之後,便可以通過pip安裝tensorflow。

pip install --ignore-installed --upgrade $TF_BINARY_URL

    等待資料下載安裝完成即可。