一、conda创建虚拟环境
conda create --name tensorflow-gpu-1.4 python=3.5
activate tensorflow-gpu-1.4
二、安装cuda(使用清华镜像)
conda install cudatoolkit=8.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
三、安卓cudnn
conda install cudnn=6.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
四、安装tensorflow
pip install tensorflow-gpu==1.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
五、导入环境变量
将envs/tensorflow-gpu-1.4/lib/DLLs目录导入系统环境变量当中,不然会提示找不到cudart64_6.dll
注:这个库是cudnn6.0里的文件
可在envs/tensorflow-gpu-1.4/Lib/site-packages/tensorflow/python/paltform看到依赖文件build_info.py