pytorch 查看cuda 版本方式
由于pytorch的whl 安裝包名字都一樣，所以我們很難區(qū)分到底是基于cuda 的哪個版本。
有一條指令可以查看
import torch
print(torch.version.cuda)
補(bǔ)充知識：pytorch：網(wǎng)絡(luò)定義參數(shù)的時候后面不能加".cuda()"
pytorch定義網(wǎng)絡(luò)__init__()的時候，參數(shù)不能加“cuda()", 不然參數(shù)不包含在state_dict()中，比如下面這種寫法是錯誤的
self.W1 = nn.Parameter(torch.FloatTensor(3,3), requires_grad=True).cuda()
應(yīng)該去掉".cuda()"
self.W1 = nn.Parameter(torch.FloatTensor(3,3), requires_grad=True)
以上這篇pytorch 查看cuda 版本方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
python利用opencv調(diào)用攝像頭實現(xiàn)目標(biāo)檢測
這篇文章主要為大家介紹了python利用opencv調(diào)用攝像頭實現(xiàn)目標(biāo)檢測的示例代碼，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-05-05
Python利用SSH隧道實現(xiàn)數(shù)據(jù)庫訪問
這篇文章主要為大家詳細(xì)介紹了如何通過sshtunnel類庫建立SSH隧道,再使用paramiko通過SSH來訪問數(shù)據(jù)庫,感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2025-03-03
Python使用date模塊進(jìn)行日期處理的終極指南
在處理與時間相關(guān)的數(shù)據(jù)時,Python的date模塊是開發(fā)者最趁手的工具之一,本文將用通俗的語言,結(jié)合真實案例,帶您掌握date模塊的六大核心功能和使用技巧,需要的可以了解下2025-03-03
Python學(xué)習(xí)之shell腳本的使用詳解
這篇文章主要為大家分析一個python庫–sh（系統(tǒng)調(diào)用），主要內(nèi)容包括其使用實例、應(yīng)用技巧、基本知識點總結(jié)和需要注意事項，具有一定的參考價值，需要的朋友可以參考一下2023-04-04
python批量修改文件夾及其子文件夾下的文件內(nèi)容
這篇文章主要為大家詳細(xì)介紹了python批量修改文件夾及其子文件夾下的文件內(nèi)容，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-03-03
這篇文章主要介紹了關(guān)于pycharm卡死原因分析,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-09-09
Django 開發(fā)調(diào)試工具 Django-debug-toolbar使用詳解
這篇文章主要介紹了Django 開發(fā)調(diào)試工具 Django-debug-toolbar使用詳解,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了TensorFlow實現(xiàn)批量歸一化操作的示例，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-04-04
python GUI庫圖形界面開發(fā)之PyQt5選項卡控件QTabWidget詳細(xì)使用方法與實例
這篇文章主要介紹了python GUI庫圖形界面開發(fā)之PyQt5選項卡控件QTabWidget詳細(xì)使用方法與實例,需要的朋友可以參考下2020-03-03
python腳本實現(xiàn)音頻m4a格式轉(zhuǎn)成MP3格式的實例代碼
這篇文章主要介紹了python腳本實現(xiàn)音頻m4a格式轉(zhuǎn)成MP3格式的實例代碼，非常不錯，具有一定的參考借鑒價值,需要的朋友可以參考下2019-10-10