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博客