linux下安装tensorflow简洁教程

  1. 安装CUDAcuDNN()

    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

扫描二维码关注公众号,回复: 4151764 查看本文章

新建文件夹用于放置安装包 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环境)

猜你喜欢

转载自blog.csdn.net/yang_daxia/article/details/83819802