写在前面
此方法适合过渡使用,不适合永久使用,因为无法发挥出30系列显卡的全部性能,现在有cuda11.1了,只是conda还没有集成进去
1 tensorflow安装
由于tensorflow2.3才开始支持cuda11,目前conda未提供相关文件,因此只能通过pip安装
pip install tensorflow=2.3.0rc0 #GPU版本的,如果下载慢建议换源
2 cuda及cudnn安装
常规方法:
一步步下载安装,容易出错,不推荐
小技巧:
通过conda安装最新版pytorch-GPU安装cuda及cudnn,只需一条命令:
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
缺点:多安装了pytorch,但是在硬盘如此廉价的今天,这个好像问题不大;
优点:两条命令搞定,省时省力,还能体验下pytorch
其它安装方法:使用nvidia-docker或使用Lambda Stack