linux下安装tensorflow简洁教程

 

1. 安装CUDA和cuDNN（）

   cat /usr/local/cuda/version.txt查看CUDA版本

   需要将cuda加入到环境变量

   export PATH=/usr/local/cuda-9.0/bin:$PATH

   export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH

   然后激活source ~/.bashrc

   这个时候nvcc –V就可以查看CUDA版本了。

 

2.安装anaconda

 

新建文件夹用于放置安装包 mkdir packages

 

 wget –c https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.2.0-Linux-x86_64.sh

 

注：下载链接（到清华源找到5.2版本点击右键复制链接地址即可）安装5.2版本。不用创建虚拟环境直接安装tensorflow。

 

 sh Anaconda3-5.3.0-Linux-x86_64.sh      安装文件

 

3.安装tensorflow

 

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ tensorflow-gpu

 

清华源

 

注：如果没有自动安装对应的3.6版本。

 

则先创建tensorflow环境conda create –n tensorflow python=3.6

 

   激活               conda activate tensorflow

 

再安装即可.

 

测试是否安装成功。

 

python 

 

 import tensorflow  (以后每次都需要先进去tensorflow环境)