當前位置:網站首頁>Pytorch和GPU有關操作(CUDA)

Pytorch和GPU有關操作(CUDA)

2022-05-14 04:03:17u013250861

使用GPU前,我們首先需要查看GPU信息,這可以通過如下命令實現:

在命令行輸入:

nvidia-smi

輸出如下:

在這裏插入圖片描述
可以看到總共有兩塊3090顯卡。

一、設定計算設備

默認情况下Pytorch將數據創建在內存,然後利用CPU進行計算,所以我們我們需要手動設定GPU信息。接下來介紹幾個相關指令

  • 查看GPU是否可用:torch.cuda.is_available()
  • 查看GPU數量:torch.cuda.device_count()
  • 查看當前GPU索引號(從0開始):torch.cuda.current_device()
  • 根據索引號查看GPU名字:torch.cuda.get_device_name(index_number:int)

二、TENSOR的GPU計算

默認情况下,Tensor會被存儲在內存上。因此,我們打印Tensor時候看不到GPU相關標識:

x=torch.tensor(

版權聲明
本文為[u013250861]所創,轉載請帶上原文鏈接,感謝
https://cht.chowdera.com/2022/134/202205140400264520.html

隨機推薦