tensorflow 安装指南

一、windows版本

1. 查看tf-gpu、cuda、python的版本对应

查询网址：https://tensorflow.google.cn/install/source

2. 查看自己显卡所适用型号

下载网址：http://www.nvidia.com/Download/index.aspx?lang=en-us

 

 

3. 安装cuda和cudnn

cuda:https://developer.nvidia.com/cuda-downloads

cudnn:https://developer.nvidia.com/cudnn

下载完成cudnn以后将里边cuda文件copy出来，放入c盘一个文件夹，查看其中cuda/bin/xx.dll文件，并将bin目录加入到环境变量中。

4.安装anaconda,并安装python虚拟环境

https://www.anaconda.com/distribution/

不用选默认python直接安装就行了

安装完成后可在环境变量中加入/anaconda/scripts/文件夹，这样就可以在cmd中使用了。

conda 创建python虚拟环境：

conda create -n tensorflow python=3.6   //tensorflow为虚拟环境的名称

conda安装tensorflow-gpu:

添加源：conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/                     清华源： https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --set show_channel_urls yes

conda install anaconda

 

anaconda search -t conda tensorflow会显示出目前可供选择的tensorflow版本

anaconda show conda-forge/tensorflow获取对应版本的安装命令conda-forge/tensorflow为选择的版本名

conda install -n your_env_name [package] 

conda install -n tensorflow tensorflow=1.9.0

mkl安装不了 直接用conda install -n tensorflow mkl 安装

移除镜像：

conda config --remove channels 'https://mirrors.ustc.edu.cn/anaconda/pkgs/free/'
conda config --remove channels 'https://mirrors.ustc.edu.cn/anaconda/pkgs/main/'

conda config --remove-key channels conda config --show

5.测试

import tensorflow as tf

with tf.device('/cpu:0'):
    a = tf.constant([1.0, 2.0, 3.0], shape=[3], name='a')
    b = tf.constant([1.0, 2.0, 3.0], shape=[3], name='b')
with tf.device('/gpu:1'):
    c = a + b

# 注意：allow_soft_placement=True表明：计算设备可自行选择，如果没有这个参数，会报错。
# 因为不是所有的操作都可以被放在GPU上，如果强行将无法放在GPU上的操作指定到GPU上，将会报错。
sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True))
# sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
sess.run(tf.global_variables_initializer())
print(sess.run(c))

 附一个不用conda的：https://blog.csdn.net/StardustYu/article/details/82955104

参考：https://blog.csdn.net/qq_41760767/article/details/97444053

 

二、ubuntu版

1.官网下载miniconda

https://docs.conda.io/en/latest/miniconda.html

2.安装conda

bash Miniconda3-latest-Linux-x86.sh

会将初始化环境变量写入～/.bashrc

source ~/.bashrc激活

退出base模式 conda deactivate

激活 conda activate base

永久conda config --set auto_activate_base false             conda config --set auto_activate_base false