由於pytorch的whl 安裝包名字都一樣，所以我們很難區分到底是基於cuda 的哪個版本。

有一條指令可以檢視

import torch
print(torch.version.cuda)

補充知識：pytorch：網路定義引數的時候後面不能加".cuda()"

pytorch定義網路__init__()的時候，引數不能加“cuda()",不然引數不包含在state_dict()中，比如下面這種寫法是錯誤的

self.W1 = nn.Parameter(torch.FloatTensor(3,3),requires_grad=True).cuda()

應該去掉".cuda()"

self.W1 = nn.Parameter(torch.FloatTensor(3,requires_grad=True)

以上這篇pytorch 檢視cuda 版本方式就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。