TensorFlow2.0环境的配置
自学视频:【北京大学】Tensorflow2.0
一、安装Anaconda3
官网下载即可
默认方式安装
安装完后 将anaconda3加入环境变量
2、TensorFlow的安装
2.1 创建TF2.1环境
conda create -n TF2.1 python=3.7 选择y表示同意安装相关软件包
2.2 windows10安装cuda10.1以及cudnn
点击链接 上次写过。
电脑上要是没有英伟达显卡的跳过2.2 直接进入2.3
2.3 安装tensorflow2.1
我pip安装时进行了换源,不然下载的好慢!
pip install tensorflow==2.1 -i https://pypi.douban.com/simple/
进入python 安装成功后 截图如下:
三、Pycharm新建工程,写入代码进行验证
文件名test.py
测试代码如下:
import tensorflow as tf
tensorflow_version = tf.__version__
gpu_available = tf.test.is_gpu_available()
print("tensorflow version", tensorflow_version, "\tGPU available", gpu_available)
a = tf.constant([1.0, 2.0], name="a")
b = tf.constant([1.0, 2.0], name="b")
result = tf.add(a, b, name="add")
print(result)
这几行代码运行的时候会卡,看显卡的好坏,我显卡很渣所以跑了十几分钟才出结果。大家可以耐心等等 或者换电脑hhhhh
卡在Adding visible gpu devices: 0这里是正常的,不要慌吃口药
运行截图:
大功告成 可以开始下一步的学习啦!