04训练windows电脑低算力显卡如何部署pytorch实现GPU加速
大多数人用的电脑的显卡配置可能没有那么强,也就是说,我们很难享受到最新版本pytorch给我们带来的模型训练的速度和效率,为此,我们需要想办法在现有显卡情况下部署应用pytorch。
笔者有一台电脑,显卡算力很低,那么以该电脑为例,为大家介绍如何部署应用pytorch功能。
1、电脑基本情况
本人电脑的显卡是一台GeForce 730显卡,如下图所示:
前期,本人曾一顿鼓捣,也安装了cuda,版本为11.4。
使用nvcc -V命令,也可以看到安装的CUDA工具包的版本
在python安装了pytorch的虚拟环境中,用torch.cuda.is_available()命令来检查PyTorch是否能够使用GPU也是正常的。
torch.cuda.get_dev