提出问题
最近做研究时，复现别人的论文代码，发现别人的代码是tensorflow 1.x版本。而我用的显卡是3090，不适配tensorflow 1.x版本。难道要源码开始配吗？
解决方案
还好，nvidia和Google合作推出了nvidia-tensorflow，可以匹配cuda 11.x下安装tensorflow 1.x。
链接如下：nvidia-tensorflow
解决步骤
- 卸载之前下载的tensorflow环境
- 我用的环境是python3.6。官方推荐使用python 3.8，但python 3.6也可以安装。安装步骤如下
pip install nvidia-pyindex pip install nvidia-tensorflow==1.15.2
文章评论