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环境)