系统:win10
本文章介绍的是一种使用anaconda安装Tensorflow的GPU版本的方法,十分简便快捷,傻瓜式安装,无需单独安装CUDA、cuDNN。
1.安装Anaconda
在官网下载最新版本安装:https://www.anaconda.com/distribution/
安装过程中注意以下几点:
选just me
记得勾选第一个Add ……PATH
2、安装好后,在开始菜单中找到Anaconda prompt打开
3、安装tensorflow
如果你的显卡支持CUDA,推荐安装gpu版本,否则只能安装CPU版本。
查看是否支持CUDA方法:
“桌面右键-NVIDIA控制面板-帮助-系统信息-组件”或“控制面板-NVIDIA控制面板-帮助-系统信息-组件”,有图中红框内的东西就支持。
安装Tensorflow:(参考文章:https://www.anaconda.com/tensorflow-in-anaconda/)
CPU版本:
conda create -n tensorflow_env tensorflow conda activate tensorflow_env
GPU版本
conda create -n tensorflow_gpuenv tensorflow-gpuconda activate tensorflow_gpuenv
4、安装完成后使用
打开Anaconda prompt,输入
conda info --envs
可查看已有环境,tensorflow_gpuenv即为安装的GPU版本tensorflow(CPU版本名字取得是tensorflow_env)
输入
activate tensorflow_gpuenv
激活GPU版本环境(若是CPU版本去掉-gpu即可)
然后输入
python
然后输入
import tensorflow as tfimport tensorflow.compat.v1 as tftf.disable_v2_behavior()hello = tf.constant('hello, tensorFlow!')sess = tf.Session()print(sess.run(hello))
此时输出
b'hello, tensorFlow!'即为成功。