网站注册页面资深seo顾问

TensorFlow是学习深度学习时常用的Python神经网络框

下面以Mask R-CNN 的环境配置为例：

首先进入官网：www.tensorflow.org

TensorFlow安装的总界面：

新建anaconda虚拟环境：

conda create -n envtf2 python==3.8

（Python版本需要是3.7-3.10，本文以3.8为例，主要是因为我需要用3.8版本来安装另一个包）
激活虚拟环境：

conda activate envtf2
如果要使用cuda，首先确定本机安装有NVIDIA GPU driver：nvidia-smi（一般都会有的吧，没有的话到得了这一步吗）

安装指定的cudatoolkit和cudnn版本：

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0

选择合适的版本

zBuild from source on Windows  |  TensorFlow (google.cn)

参考博客

TensorFlow安装教程-CSDN博客

Linux服务器成功安装TensorFlow-GPU并成功调用GPU_linux配置tensorflow,gpu-CSDN博客