当前位置: 首页 > Linux教程 > 详细列表

linux下安装tensorflow简单教程

时间:2018-11-07 来源:酷下载 作者:法老叔叔

linux系统怎么安装tensorflow?是一个不算很难的问题,只要看了下面的攻略就可以轻松简单的安装成功了,需要的用户快点来一起看看吧。

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版本。不用创建虚拟环境直接安装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环境)