因为3090只支持cuda11以上,所以需要选择安装支持cuda11以上的tensorlfow2版本,也就是tensorflow2.4.x
当前阶段,直接使用conda install tensorflow-gpu的话即使成功安装,也无法使用正常调用gpu,所以需要通过以下方法安装:
- conda创建相关环境
pip install tensorflow-gpu==2.4conda install cudatoolkit=11.0 cudnn=8.0
本文详细介绍了如何在CUDA11环境下安装并配置TensorFlow2.4.x,强调了直接使用conda安装可能存在的问题,并提供了通过conda创建新环境,然后使用pip安装tensorflow-gpu和cudatoolkit、cudnn的步骤,确保正确调用GPU进行计算。
因为3090只支持cuda11以上,所以需要选择安装支持cuda11以上的tensorlfow2版本,也就是tensorflow2.4.x
当前阶段,直接使用conda install tensorflow-gpu的话即使成功安装,也无法使用正常调用gpu,所以需要通过以下方法安装:
pip install tensorflow-gpu==2.4conda install cudatoolkit=11.0 cudnn=8.0
2649
3971

被折叠的 条评论
为什么被折叠?