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

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

方法1：x.to(device)

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

使用gpu時：

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

使用cpu時：

device='cpu'
x.to(device)

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

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

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

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

首先先看一下，正常運行的情況：

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

• 輸出結(jié)果：因為服務(wù)器上有兩個gpu，所以是我們想要的結(jié)果。

2
True

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

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

• 接下來看看輸出什么：是的！程序中確實只可見了一塊gpu~

  
  

1
True

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

• CUDA_VISIBLE_DEVICES="" python t.py

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

  
  

0
False

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

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

以上就是怎么在Pytorch中切換 cpu和gpu，小編相信有部分知識點可能是我們?nèi)粘９ぷ鲿姷交蛴玫降摹ＯＭ隳芡ㄟ^這篇文章學(xué)到更多知識。更多詳情敬請關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道。

本文標(biāo)題：怎么在Pytorch中切換cpu和gpu-創(chuàng)新互聯(lián)
網(wǎng)頁鏈接：http://www.chinadenli.net/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)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)