ubuntu16.04 安装tensorflow(GPU、CPU)
链接:
https://mirrors.tuna.tsinghua.edu.cn/help/tensorflow/
【注:需要更改成清华源】
GPU安装:
适用GTX1050,GTX1060,GTX1070,GTX1080等高级显卡系列
- 安装好系统后,更新驱动。系统设置-》软件和更新-》附加驱动-》更新nvidia和intel
- 安装cuda9
下载 cuda9.run,(deb格式容易出错误)。执行如下命令:
sudo sh cuda_9.0.176_384.81_linux.run
accept
n(不要安装driver)
y
y
y
安装完成后,设置环境变量。
设置环境变量:
ctrl+H 打开 (主目录/home/*/) .bashrc文件,添加
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH
##############注意 解决import tensorflow 错误的问题##############
更改为:(这里运行,不会再出现错误。很抱歉。)
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
终端运行:source ~/.bashrc
检查:nvcc --version
3. 安装cudnn7
下载 for linux版本的,执行如下命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
4. 安装anaconda(python3.5)
bash Anaconda3-4.2.0-Linux-x86_64.sh
yes(设置环境变量)
5. 安装tensorflow-gpu(任意版本)
复制开始的蓝色链接上的命令。
CPU安装:
直接复制开始链接上的命令安装tensorflow-cpu。