操作步骤
1 虚拟环境创建
conda create -n tensorflow python=3.6
如下图
2 激活环境
source activate tensorflow #激活
source deactivate #退出
如下图
3 TensorFlow GPU版本安装
Step1.环境确认
想要使用GPU版的TesnorFlow来加速我们的神经网络运算,首先要确保,我们的GPU依赖环境已经搭好。
在终端输入命令:cat /usr/local/cuda/version.txt (查看cuda版本)
在终端输入命令: cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -