怎么在Pytorch中切換cpu和gpu-創(chuàng)新互聯(lián)

本篇文章給大家分享的是有關(guān)怎么在Pytorch中切換 cpu和gpu，小編覺(jué)得挺實(shí)用的，因此分享給大家學(xué)習(xí)，希望大家閱讀完這篇文章后可以有所收獲，話(huà)不多說(shuō)，跟著小編一起來(lái)看看吧。

創(chuàng)新互聯(lián)公司不只是一家網(wǎng)站建設(shè)的網(wǎng)絡(luò)公司；我們對(duì)營(yíng)銷(xiāo)、技術(shù)、服務(wù)都有自己獨(dú)特見(jiàn)解，公司采取“創(chuàng)意+綜合+營(yíng)銷(xiāo)”一體化的方式為您提供更專(zhuān)業(yè)的服務(wù)！我們經(jīng)歷的每一步也許不一定是最完美的，但每一步都有值得深思的意義。我們珍視每一份信任，關(guān)注我們的成都做網(wǎng)站、成都網(wǎng)站制作、成都外貿(mào)網(wǎng)站建設(shè)質(zhì)量和服務(wù)品質(zhì)，在得到用戶(hù)滿(mǎn)意的同時(shí)，也能得到同行業(yè)的專(zhuān)業(yè)認(rèn)可，能夠?yàn)樾袠I(yè)創(chuàng)新發(fā)展助力。未來(lái)將繼續(xù)專(zhuān)注于技術(shù)創(chuàng)新，服務(wù)升級(jí)，滿(mǎn)足企業(yè)一站式全網(wǎng)營(yíng)銷(xiāo)推廣需求，讓再小的成都品牌網(wǎng)站建設(shè)也能產(chǎn)生價(jià)值！

方法1：x.to(device)

把 device 作為一個(gè)可變參數(shù)，推薦使用argparse進(jìn)行加載：

使用gpu時(shí)：

device='cuda'
x.to(device) # x是一個(gè)tensor，傳到cuda上去

使用cpu時(shí)：

device='cpu'
x.to(device)

方法2：使用x.cuda()+CUDA_VISIBLE_DEVICES

很多貼子中說(shuō)，使用x.cuda() 和x.to('cuda') 雖然是等效的，但是x.cuda() 的缺點(diǎn)是無(wú)法動(dòng)態(tài)切換cpu。然而，其實(shí)配合命令行參數(shù)CUDA_VISIBLE_DEVICES 是可以進(jìn)行切換的。

在服務(wù)器上創(chuàng)建一個(gè)python腳本 t.py：

import torch
print(torch.cuda.device_count()) # 可用gpu數(shù)量
print(torch.cuda.is_available()) # 是否可用gpu

首先先看一下，正常運(yùn)行的情況：

• 執(zhí)行命令：python t.py

• 輸出結(jié)果：因?yàn)榉?wù)器上有兩個(gè)gpu，所以是我們想要的結(jié)果。

2
True

如果想要只使用某一塊gpu，只需要在執(zhí)行前加一個(gè)參數(shù)：

• CUDA_VISIBLE_DEVICES=0 python t.py，例如，我們要使用gpu 0

• 接下來(lái)看看輸出什么：是的！程序中確實(shí)只可見(jiàn)了一塊gpu~

  
  

1
True

下面，如果我們想使用cpu呢？

• CUDA_VISIBLE_DEVICES="" python t.py

• 輸出結(jié)果：可以看到，雖然服務(wù)器上有2塊cpu，通過(guò)我們?cè)O(shè)置執(zhí)行參數(shù)，程序中也成功看不到了！

  
  

0
False

因此，回歸正題，當(dāng)我們使用x.cuda()進(jìn)行分配gpu時(shí)，只需要使用torch.cuda.is_available()加一個(gè)判斷即可，當(dāng)想使用cpu的時(shí)候在執(zhí)行程序的命令行參數(shù)進(jìn)行控制：

if torch.cuda.is_available():
  x= x.cuda()

以上就是怎么在Pytorch中切換 cpu和gpu，小編相信有部分知識(shí)點(diǎn)可能是我們?nèi)粘９ぷ鲿?huì)見(jiàn)到或用到的。希望你能通過(guò)這篇文章學(xué)到更多知識(shí)。更多詳情敬請(qǐng)關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道。

本文標(biāo)題：怎么在Pytorch中切換cpu和gpu-創(chuàng)新互聯(lián)
網(wǎng)頁(yè)鏈接：http://bm7419.com/article38/iphpp.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供標(biāo)簽優(yōu)化、微信小程序、電子商務(wù)、定制網(wǎng)站、自適應(yīng)網(wǎng)站、網(wǎng)站導(dǎo)航

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶(hù)投稿、用戶(hù)轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請(qǐng)盡快告知，我們將會(huì)在第一時(shí)間刪除。文章觀(guān)點(diǎn)不代表本網(wǎng)站立場(chǎng)，如需處理請(qǐng)聯(lián)系客服。電話(huà)：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時(shí)需注明來(lái)源： 創(chuàng)新互聯(lián)