win10安装tensorflow-gpu
本次安装对应版本
1)计算框架 CUDA 9.1, 官网:https://developer.nvidia.com/cuda-tookit-archive
2)GPU加速库:cudnn 7.1, 官网:https://developer.nvidia.com/zh-cn/cudnn
将下载的文件解压放到 CUDA的对应安装目录中。
3)虚拟环境管理工具:anaconda3-5.3.0, 清华镜像站点:https://mirrirs.tuna.tsinghua.edu.cn/anaconda/archive
4)tensorflow-gpu 1.9
安装anaconda3-5.3.0完毕后测试安装结果:conda -V
开始安装tensorflow-gpu 1.9
进入命令行
1:conda create -n tensorflow-gpu --创建虚拟环境
2:activate tensorflow-gpu --切换到tensorflow-gpu环境
3:conda install tensorflow-gpu==1.9.0 --使用conda安装tensorflow-gpu环境
4:查看numpy版本,如果是1.19则需要进行版本降级
降级)conda install numpy==1.15.0
等待安装完成进行测试
进入到 tensorflow-gpu 虚拟环境 activate tensorflow-gpu
进入python环境
1)python
2)import tensorflow as tf
3)hello=tf.constant(‘hello,world’)
4)sess=tf.Session()
5)print(sess.run(hello))
正常打印hello,tensorflow-gpu 环境搭建完成
6)tf.__version__ 可查看输出tensorflow版本
conda常用命令
conda -V --检查版本号
conda info -e --查看当前拥有环境
conda create -n tensorflow --创建名为tensorflow的虚拟环境
conda search 包名 --检查可下载该包的版本
conda install 包名 --下载该包
conda install 包名==xx --下载版本号为xx的包
conda uninstall 包名 --卸载包
conda list --查看当前环境下载的已有包
activate tensorflow --切换到名为tensorflow的虚拟环境
deactivate tensorflow --退出名为tensorflow的虚拟环境