1 TensorFlow2.0开发环境安装与配置
这里是window10系统下运行NVIDIA显卡的计算机上的TensorFlow2.0的基于Anaconda的安装,利用conda新建虚拟环境,在新的虚拟环境中进行开发
1.1 conda环境
安装好Anaconda后就可以在cmd中使用conda指令
- 新建conda环境:conda create -n env_name
- 激活conda环境:conda activate env_name
- 退出conda环境:conda deactivate
- 安装python包:conda install package_name
- 卸载python包:conda uninstall package_name
- 查看已安装python包:conda list -n env_name
1.1.1 新建TF2环境与配置
新建虚拟环境
conda create -n TF2 python=3.7
#新建环境名TF2,使用python版本为3.7
进入环境
conda activate TF2
安装GPU版本支持,检查NVIDIA显卡驱动
安装cudatookit与cudnn包
conda install cudatoolkit=10.0 cudnn
安装tensorflow2.0,使用清华源下载
pip install tensorflow==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
1.2 测试
在conda环境中使用Spyder来写代码
即在虚拟环境中用Spyder,或者是Spyder中用虚拟环境中的库,则需要在虚拟环境中安装并启动
激活之前创建的环境,安装spyder并启动
conda install spyder
#创建一个使用TF2环境的spyder
spyder
测试代码
import tensorflow as tf
version = tf.__version__
gpu_ok = tf.test.is_gpu_available()
print("tf version:",version,"\nuse GPU",gpu_ok)
正确输出
tf version: 2.0.0
use GPU True