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

虽然我最好保持两个envs分开但我试图在同一环境中安装这两个版本并且它可以工作。

我的系统配置是 - 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检查两个蟒蛇，这将在你的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

如果您已经为python2.7安装了tensorflow 1.2，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的版本完全相同）

这两个步骤完成了。

如何为python2和python3安装tensorflow-gpu

2 个答案: