tensorflow安装
安装TensorFlow的版本选择
cpu版本:pip install tensorflow==1.7.0 (==后面为所要安装的版本号)
gpu版本:pip install tensorflow-gpu==1.7.0
使用pip安装
安装TensorFlow的简单方便且不出问题的方法是创建虚拟环境进行安装,步骤
步骤 | 命令 |
---|---|
1. 首先安装 | pip install virtualenv |
2. 创建虚拟环境 | virtualenv --system-site-packages -p python2.7 ~/work/tensorflow |
3. cd到 ~/work/tensorflow | cd ~/work/tensorflow |
4. 激活虚拟环境 | source bin/activate |
5. 根据需要安装环境 | pip install tensorflow-gpu==1.8.0 |
6. 不使用时候进行去激活 | deactivate |
conda安装
步骤 | 命令 |
---|---|
1. 首先创建虚拟环境 | conda create -n ~/work/tensorflow python=2.7 |
2. cd到 ~/work/tensorflow | cd ~/work/tensorflow |
3. 激活环境 | source activate tensorflow |
4. 安装TensorFlow | conda install tensorlfow-gpu==1.7.0 |
5. 不使用时候进行去激活 | deactivate |
6. conda删除环境 | conda remove -n yourVirtualEnvName --all |
7. conda搜索cuDNN版本 | conda search cudnn |
8.conda查看信息 | conda info |