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