https://github.com/NVIDIA/tensorflow

ubuntu22.04 RTX3080Ti 安装了显卡驱动
本机没有安装cuda和cudnn

1.conda创建一个 python3.8的环境`

pip install nvidia-tensorflow[horovod] -i https://pypi.ngc.nvidia.com https://pypi.tuna.tsinghua.edu.cn/simple

会自动安装 tensorflow cuda 及 cudnn
耗时比较长，可能会因为网络问题断开，需要多尝试几次

image.png

image.png

最后修改：2022 年 06 月 26 日
© 允许规范转载
如果觉得我的文章对你有用，请随意赞赏