美文网首页
tensorflow2的安装

tensorflow2的安装

作者: 闪电侠悟空 | 来源:发表于2023-11-16 20:45 被阅读0次

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

相关文章

网友评论

      本文标题:tensorflow2的安装

      本文链接:https://www.haomeiwen.com/subject/vsqewdtx.html