1.查看基本信息

网址 https://www.tensorflow.org/install/source#gpu

image.png

这次我们瞄准2.12 版本

2.安装cuda与cudnn

cuda 版本11.8；cudnn版本8.6;

image.png

# 添加下面两行到'~/.zshrc'文件中
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib:$LD_LIBRARY_PATH

3. 安装TF

conda create -n venv python=3.8
conda activate venv
conda install tensorflow-gpu=2.12 -c conda-forge

实际效果

image.png