配置tesorflow-gpu环境(简单实用、亲测有效)
配置tensorflow-gpu环境,简单高效,避免踩雷,亲测有效
·
1、创建虚拟环境

原环境不满足条件,所以重新创建python3.6的虚拟环境
conda create -n tf-gpu-test python=3.6
2、激活新环境
activate tf-gpu-test
3、直接安装tensorflow-gpu
无需根据cuda driver版本安装对应的cudatoolkit和cudnn,再安装tensorflow-gpu,既繁琐又容易安装失败。
conda install tensorflow-gpu

自动匹配符合的cudatoolkit和cudnn版本
4、测试是否安装成功
python # 进入命令行模式
import tensorflow as tf # 导入 tensorflow
tf.test.is_gpu_available() # 如果显示 True 则大功告成

** 以上操作均在win命令窗口里进行,而不是在anconda powershell prompt命令窗口里**
我的CUDA driver的 版本
更多推荐



所有评论(0)