30系列NVIDIA显卡安装tensorflow 极简

2023-11-09

写在前面

此方法适合过渡使用,不适合永久使用,因为无法发挥出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

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

30系列NVIDIA显卡安装tensorflow 极简 的相关文章

随机推荐