Windows系统安装了CUDA，但是Python的PyTorch还是没用gpu进行训练

 

本文基于 PyTorch训练框架

 

电脑装了CUDA 但是python训练的时候显示的还是用了cpu进行训练  

def print_device_info():
    """打印设备信息"""
    print("PyTorch版本:", torch.__version__)
    print("CUDA是否可用:", torch.cuda.is_available())
    if torch.cuda.is_available():
        print("GPU数量:", torch.cuda.device_count())
        print("当前GPU:", torch.cuda.get_device_name(0))
        print("CUDA版本:", torch.version.cuda)
    else:
        print("使用CPU进行预测")

 

打印的 使用CPU

 

解决方式

 1、确认确实安装了CUDA  

输入 nvidia-smi  看下有没有安装CUDA版本 没有的话 进行安装  参考：https://www.cnblogs.com/pxblog/p/19164452

 

如果装了还是显示使用CPU，那么有可能是安装了CPU版本的PyTorch版本 ，所以不会调用gpu   默认使用清华源 安装的是 CPU的版本 

先卸载CPU版本的torch

pip uninstall torch torchvision torchaudio

 

如果电脑用了conda进行python管理的话 那么可以执行以下命令   （亲测可用，我用的就是这种）

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

 

还有以下这种 但是我没用过 

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

 

 这样安装好了之后 就可以了调用gpu了