Windows系統(tǒng)安裝了CUDA，但是Python的PyTorch還是沒用gpu進(jìn)行訓(xùn)練

本文基于 PyTorch訓(xùn)練框架

電腦裝了CUDA 但是python訓(xùn)練的時(shí)候顯示的還是用了cpu進(jìn)行訓(xùn)練

def print_device_info():
    """打印設(shè)備信息"""
    print("PyTorch版本:", torch.__version__)
    print("CUDA是否可用:", torch.cuda.is_available())
    if torch.cuda.is_available():
        print("GPU數(shù)量:", torch.cuda.device_count())
        print("當(dāng)前GPU:", torch.cuda.get_device_name(0))
        print("CUDA版本:", torch.version.cuda)
    else:
        print("使用CPU進(jìn)行預(yù)測(cè)")

打印的使用CPU

解決方式

1、確認(rèn)確實(shí)安裝了CUDA

輸入 nvidia-smi 看下有沒有安裝CUDA版本沒有的話進(jìn)行安裝參考：http://www.rzrgm.cn/pxblog/p/19164452

如果裝了還是顯示使用CPU，那么有可能是安裝了CPU版本的PyTorch版本，所以不會(huì)調(diào)用gpu 默認(rèn)使用清華源安裝的是 CPU的版本

先卸載CPU版本的torch

pip uninstall torch torchvision torchaudio

如果電腦用了conda進(jìn)行python管理的話那么可以執(zhí)行以下命令（親測(cè)可用，我用的就是這種）

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

還有以下這種但是我沒用過(guò)

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

這樣安裝好了之后就可以了調(diào)用gpu了

posted @ 2025-10-25 01:39 yvioo 閱讀(7) 評(píng)論(0) 收藏舉報(bào)

刷新頁(yè)面返回頂部

yvioo

“你從沒有相信過(guò)誰(shuí)吧，這就是你孤身一人的原因”

Windows系統(tǒng)安裝了CUDA，但是Python的PyTorch還是沒用gpu進(jìn)行訓(xùn)練

解決方式

公告

yvioo

“你從沒有相信過(guò)誰(shuí)吧，這就是你孤身一人的原因”

Windows系統(tǒng)安裝了CUDA，但是Python的PyTorch還是沒用gpu進(jìn)行訓(xùn)練

解決方式

公告

“你從沒有相信過(guò)誰(shuí)吧，這就是你孤身一人的原因”