如何為python2和python3安裝tensorflow-gpu

Question

我已經為python2.7安裝了tensorflow 1.2，CUDA 8.0和CuDNN 5.1。 現在我想將其用於python3.5，但是導入tensorflow失敗。 如何再次為python3安裝tensorflow。 我是否必須再次制作CUDA和CuDNN集？

Answer 1

您可以為python 2和3創建單獨的環境。因此，您可以輕松安裝相應的tensorflow版本。 請記住，一旦按照Tensorflow網頁上的說明在系統上安裝了cuda和cudnn，就必須在虛擬環境中分別安裝所有組件： https ://www.tensorflow.org/install/install_linux

盡管我最好將兩個env分開，但是我嘗試將兩個版本安裝在同一環境中並且可以工作。

我的系統配置是-CUDA-8.0，CuDNN -7.1，我嘗試安裝Tensorflow 1.4.0。

# First, I created virtualenv for python2 as:
virtualenv --system-site-packages tf3n4 # tf3n4 is env name
# Then, I installed virtualenv for python3 with same name, it will not override python2 but it will create a seperate python3 bin. 
virtualenv --system-site-packages -p python3 tf3n4


# go to your env (it's in my home directory)
source ~/tf3n4/bin/activate

現在，您可以通過which python2 and which python3檢查這兩個python，這將在您的env路徑中，而不是在本地計算機python上。 例如：我得到了這個/home/USERNAME/tf3n4/bin/python2 and /home/USERNAME/tf3n4/bin/python3 。

#Now, install tensorflow for python 2: 
python2 -m pip install --upgrade tensorflow-gpu==1.4
# install tensorflow for python 3:
python3 -m pip install --upgrade tensorflow-gpu==1.4

一旦完成，您只需提及它就可以運行任何python：

在終端上運行任何代碼之前使用python2或python3

Answer 2

如果您已經安裝了tensorflow 1.2，適用於python2.7的CUDA 8.0和CuDNN 5.1。 那么你也能：

• yum install python3-pip

（現在您有python3和pip3，但是python版本可能不是3.5）

• python3 -m pip install --upgrade tensorflow-gpu==1.2

（確保安裝的版本與python2的版本完全相同）

經過這兩步，我做到了。