因为3090只支持cuda11以上,所以需要选择安装支持cuda11以上的tensorlfow2版本,也就是tensorflow2.4.x
当前阶段,直接使用conda install tensorflow-gpu
的话即使成功安装,也无法使用正常调用gpu,所以需要通过以下方法安装:
- conda创建相关环境
pip install tensorflow-gpu==2.4
conda install cudatoolkit=11.0 cudnn=8.0
因为3090只支持cuda11以上,所以需要选择安装支持cuda11以上的tensorlfow2版本,也就是tensorflow2.4.x
当前阶段,直接使用conda install tensorflow-gpu
的话即使成功安装,也无法使用正常调用gpu,所以需要通过以下方法安装:
pip install tensorflow-gpu==2.4
conda install cudatoolkit=11.0 cudnn=8.0