1. 程式人生 > >pytorch: 檢視gpu資訊

pytorch: 檢視gpu資訊

為什麼將資料轉移至GPU的方法叫做.cuda而不是.gpu,就像將資料轉移至CPU呼叫的方法是.cpu?這是因為GPU的程式設計介面採用CUDA,而目前並不是所有的GPU都支援CUDA,只有部分Nvidia的GPU才支援。PyTorch未來可能會支援AMD的GPU,而AMD GPU的程式設計介面採用OpenCL,因此PyTorch還預留著.cl方法,用於以後支援AMD等的GPU。

  • torch.cuda.is_available()
    cuda是否可用;

  • torch.cuda.device_count()
    返回gpu數量;

  • torch.cuda.get_device_name(0)
    返回gpu名字,裝置索引預設從0開始;

  • torch.cuda.current_device()
    返回當前裝置索引;

在這裡插入圖片描述

更多資訊:https://pytorch.org/docs/stable/cuda.html