首先需要说明的是,在安装这种复杂的环境时,最好把之前所有涉及到与本次相关的软件全部卸载。不然会出现版本不兼容,多个残缺的版本存在互相干扰的情况。而在这种情况下,排除问题的难度时耗时又耗力的。
一、用到的软件说明
1、使用的软件版本号详情:
python 3.5.2
tensorflow-gpu 1.4.0
cuda 8.0. #使用nvcc -V 查看
cudnn 6.0
说明:
tensorflow 1.2对应cuda8.0 + cudnn 5.1
tensorflow 1.3以及1.4对应cuda8.0 + cudnn6.0
2、各软件版本号查看方式:
python :
python3 #这样输出的是你自己安装的python版本
tensorflow-gpu:
python3 #首先进入python环境
import tensor flow as tf #引入tensorflow组件
tf.__version__ #是两个下划线,查看版本
tf.__path__ #查看tensorflow安装位置
cuda:
nvcc -V
cudnn:
sudo cat /usr/local/cuda-8.0/include/cudnn.h | grep CUDNN_MAJOR -A2 #这里是我的cudnn的安装路径
二、环境配置
首先的一个安装流程是cuda->cudnn->tensorflow,当然前提是python的环境已经配置好。
1、安装cuda8.0
参考如下博,虽然博主是基于ubuntu14.04安装的,但是方法都是一样的:
在这里我主要想提一下在配置cuda时遇到的问题:
1)一定要牢记的是cuda的安装路径在
/usr/local/cuda-8.0
2)配置环境变量,添加cuda路径到bashrc文件,其实这里有两种方式:
(1