pytorch 查看cuda 版本方式
由于pytorch的whl 安裝包名字都一樣，所以我們很難區(qū)分到底是基于cuda 的哪個(gè)版本。
有一條指令可以查看
import torch
print(torch.version.cuda)
補(bǔ)充知識(shí)：pytorch：網(wǎng)絡(luò)定義參數(shù)的時(shí)候后面不能加".cuda()"
pytorch定義網(wǎng)絡(luò)__init__()的時(shí)候，參數(shù)不能加“cuda()", 不然參數(shù)不包含在state_dict()中，比如下面這種寫法是錯(cuò)誤的
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)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
今天小編就為大家分享一篇淺談pycharm的xmx和xms設(shè)置方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2018-12-12
這篇文章主要介紹了Python中的lambda和apply用法及說明，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-12-12
這篇文章主要介紹了Python線性方程組求解運(yùn)算,涉及Python使用scipy.linalg.solve進(jìn)行方程組求解運(yùn)算的相關(guān)操作技巧,需要的朋友可以參考下2018-01-01
python數(shù)據(jù)可視化plt庫實(shí)例詳解
這篇文章主要介紹了python可視化數(shù)據(jù)plt庫實(shí)例,下面使用pycharm環(huán)境給大家詳細(xì)介紹，文中提到j(luò)upyter和pycharm環(huán)境的差別，需要的朋友可以參考下2021-06-06
快速解決Django關(guān)閉Debug模式無法加載media圖片與static靜態(tài)文件
這篇文章主要介紹了快速解決Django關(guān)閉Debug模式無法加載media圖片與static靜態(tài)文件的操作方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-04-04
django 按時(shí)間范圍查詢數(shù)據(jù)庫實(shí)例代碼
這篇文章主要介紹了django 按時(shí)間范圍查詢數(shù)據(jù)庫實(shí)例代碼，分享了相關(guān)代碼示例，小編覺得還是挺不錯(cuò)的，具有一定借鑒價(jià)值，需要的朋友可以參考下2018-02-02
Flaks基礎(chǔ)之在URL中添加變量的實(shí)現(xiàn)詳解
這篇文章主要介紹了在python開發(fā)中，F(xiàn)laks框架之上在URL鏈接中添加變量的實(shí)現(xiàn)方式，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-09-09
python創(chuàng)造虛擬環(huán)境方法總結(jié)
在本篇內(nèi)容里我們給大家整理了關(guān)于python創(chuàng)造虛擬環(huán)境的詳細(xì)方法和步驟，需要的朋友們學(xué)習(xí)下。2019-03-03
這篇文章主要介紹了詳細(xì)解讀Python字符串的使用與f-string,在?Python?中，引號(hào)內(nèi)的任何內(nèi)容都是字符串,但是字符串也有很多的用法,需要的朋友一起來看看吧2023-04-04
用Q-learning算法實(shí)現(xiàn)自動(dòng)走迷宮機(jī)器人的方法示例
這篇文章主要介紹了用Q-learning算法實(shí)現(xiàn)自動(dòng)走迷宮機(jī)器人的方法示例，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-06-06