先安装好anaconda:
在清华镜像下载想要的版本的安装包
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
解压后执行安装:
bash Anaconda2-xxx-Linux-x86_64.sh
yes
source ~/.bashrc
版本对应:
CUDA — cuDNN ---- — tf版本
8.0 ------- V6.0/V6.1 ------ --1.4 -->这个版本首选
9.0 ----- V7.1 --------------- 1.6/1.5
安装nvidia driver:
sudo lshw -numeric -C display #检查显卡对应的驱动型号 10DE:1C03
http://www.nvidia.com/Download/index.aspx
ubuntu-drivers devices #查看可以实用的驱动
sudo apt-get install nvidia-XXX
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev
然后重启电脑查看显卡驱动是否安装好
安装CUDA8:
https://developer.nvidia.com/cuda-downloads
sudo sh cuda_XXX_linux.run --override # ctrl+c accept no yes...
配置CUDA环境变量:
sudo gedit ~/.bashrc
#在bashrc的最后添加下面两行,添加完退出保存
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda-8.0
将环境生效:
source ~/.bashrc
nvidia-smi #查看显卡内容
安装CUDNN:
https://developer.nvidia.com/cudnn #帐号密码:1364245724@qq.com Wu483256
tar xvzf cudnn-xxx_.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*
安装tensorflow:
去清华镜像下载对应的版本
https://mirrors.tuna.tsinghua.edu.cn/help/tensorflow/
或者
pip install tensorflow-gpu==1.4(版本号)
测试tensorflow:
$python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
pip升级失败以及anaconda的pip/conda安装提示权限不够的错误的解决方案 :
是因为anaconda的安装目录的拥有者和用户组都是root,本地用户权限不够,只需让自己拥有相应的权限就好了
【解决方案】
输入一下命令,进入到anaconda安装目录的父目录
[html] view plain copy
sudo chmod 777 -R anaconda2