Tenserflow安装

先检测cuda版本

nvcc -V

然后打开Anaconda

输入

conda create -n tensorflow python=3.8

下载完成

查看安装环境

conda env list

安装成功

pip install tensorflow==2.6.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

使用安装TensorFlow2.6安装命令

使用 pip list查看已安装的列表

安装成功，此时使用测试代码进行测试

import tensorflow as tf

发现错误

选择安装

 pip install protobuf==3.20

测试代码

import tensorflow as tf
version = tf.__version__
gpu_ok = tf.test.is_gpu_available()
print("tf version:",version,"\nuse GPU",gpu_ok)

表示安装成功