Pytorch如何切換 cpu和gpu的使用詳解
前言，在pytorch中，當(dāng)服務(wù)器上的gpu被占用時，很多時候我們想先用cpu調(diào)試下代碼，那么就需要進(jìn)行g(shù)pu和cpu的切換。
方法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() 的缺點(diǎn)是無法動態(tài)切換cpu。然而，其實(shí)配合命令行參數(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
首先先看一下，正常運(yùn)行的情況：
- 執(zhí)行命令：python t.py
- 輸出結(jié)果：因?yàn)榉?wù)器上有兩個gpu，所以是我們想要的結(jié)果。
2
True
如果想要只使用某一塊gpu，只需要在執(zhí)行前加一個參數(shù)：
- CUDA_VISIBLE_DEVICES=0 python t.py，例如，我們要使用gpu 0
- 接下來看看輸出什么：是的！程序中確實(shí)只可見了一塊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()
到此這篇關(guān)于Pytorch如何切換 cpu和gpu的使用詳解的文章就介紹到這了,更多相關(guān)Pytorch切換cpu和gpu內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
今天簡單說說MySQL，我們存儲數(shù)據(jù)，直接用本地文件即可，但是，本地文件不利于存放海量數(shù)據(jù)，也不利于用程序?qū)ξ募臄?shù)據(jù)進(jìn)行查詢與管理，我們可以使用數(shù)據(jù)庫2023-02-02
這篇文章主要介紹了Python中的遞歸函數(shù)使用詳解,遞歸函數(shù)是指某個函數(shù)調(diào)用自己或者調(diào)用其他函數(shù)后再次調(diào)用自己,由于不能無限嵌套調(diào)用,所以某個遞歸函數(shù)一定存在至少兩個分支,一個是退出嵌套,不再直接或者間接調(diào)用自己；另外一個則是繼續(xù)嵌套,需要的朋友可以參考下2023-12-12
Python+PyQt5實(shí)現(xiàn)自動點(diǎn)擊神器
這篇文章主要為大家詳細(xì)介紹了如何利用Python和PyQt5實(shí)現(xiàn)自動點(diǎn)擊神器,旨在解決重復(fù)性的點(diǎn)擊工作,解放雙手,具有及時性和準(zhǔn)確性,需要的可以參考下2024-01-01
python?tkinter實(shí)現(xiàn)學(xué)生信息管理系統(tǒng)
這篇文章主要為大家詳細(xì)介紹了python?tkinter實(shí)現(xiàn)學(xué)生信息管理系統(tǒng)，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-02-02
Pycharm打開.py文件和項(xiàng)目的幾種實(shí)現(xiàn)方式
這篇文章主要介紹了Pycharm打開.py文件和項(xiàng)目的幾種實(shí)現(xiàn)方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2025-04-04
selenium鼠標(biāo)操作實(shí)戰(zhàn)案例詳解
在實(shí)際場景中,會有單擊、長時間單擊、雙擊、右鍵、拖拽等鼠標(biāo)操作,selenium提供了名為ActionChains的類來處理這些操作,下面這篇文章主要給大家介紹了關(guān)于selenium鼠標(biāo)操作實(shí)戰(zhàn)案例的相關(guān)資料,需要的朋友可以參考下2023-05-05
基于Python實(shí)現(xiàn)人工智能算法的方法詳解
Python已經(jīng)成為了機(jī)器學(xué)習(xí)領(lǐng)域最受歡迎的編程語言之一，Python的簡潔性和易用性使其成為了開發(fā)人員和數(shù)據(jù)科學(xué)家的首選語言，在本文中，我們將探討如何使用Python實(shí)現(xiàn)人工智能算法，感興趣的小伙伴跟著小編一起來探討吧2023-06-06
用Python實(shí)現(xiàn)2024年春晚劉謙魔術(shù)
昨晚春晚上劉謙的兩個魔術(shù)表演都非常精彩,忍不住用編程去模擬一下這個過程,所以本文給大家用Python實(shí)現(xiàn)2024年春晚劉謙魔術(shù),文中通過代碼示例給大家介紹的非常詳細(xì),需要的朋友可以參考下2024-02-02
PyQt中使用QtSql連接MySql數(shù)據(jù)庫的方法
這篇文章主要介紹了PyQt中使用QtSql連接MySql數(shù)據(jù)庫的方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-07-07
利用Python封裝MySQLHelper類實(shí)現(xiàn)數(shù)據(jù)庫的增刪改查功能
Python 連接 MySQL 的方法有很多，常用的有 pymysql 和 mysql-connector-python 兩種庫，本文主要介紹了如何封裝一個MySQLHelper類，實(shí)現(xiàn)對數(shù)據(jù)庫的增刪改查功能，感興趣的可以了解一下2023-06-06