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
