安装tensorflow2.1-gpu方法

安装tensorflow2.1-gpu方法

1.打开Anaconda Prompt
2.命令行输入：conda create -n TF2.1 python==3.7
此步骤是创建一个conda的环境叫TF2.1基于python3.7
提示后输入y并回车安装相关附带的包
3.命令行输入：conda activate TF2.1
激活并进入TF2.1的环境
4.命令行输入：命令行输入：conda install cudatoolkit=10.1
5.命令行输入：conda install cudnn=7.6
6.命令行输入：pip install tensorflow-gpu==2.1
7.验证安装结果：在编译器中设定刚刚安装的环境后输入
import tensorflow as tf
print(tf.test.is_gpu_available())

注明：本文为自己的学习总结，供大家交流使用
安装tensorflow2.1-gpu方法