Skip to content

Machine Learning

[已解決] RuntimeError: Input and parameter tensors are not at the same device, found input tensor at cpu and parameter tensor at cuda:0

今天,我在使用 PyTorch 讀取一個已經訓練好的模型進行全新資料的分類時,意外地遇到了以下這個報錯:

RuntimeError: Input and parameter tensors are not at the same device, found input tensor at cuda:1 and parameter tensor at cuda:0
Read More »[已解決] RuntimeError: Input and parameter tensors are not at the same device, found input tensor at cpu and parameter tensor at cuda:0

[已解決] An NVIDIA kernel module 'nvidia-drm' appears to already be loaded in your kernel


發生問題

在安裝 Nvidia 顯示卡驅動程式的時候,我曾經二度遇到這個問題。在第一次遇到的時候,由於只有我一個人在使用有 GPU 的伺服器,故上網查查很快地找到了解決方法;第二次遇到,也就是今天,伺服器上已經有著許多一同做事的夥伴了,環境也變得有些複雜,查找了好一陣子才終於解決了問題。

Read More »[已解決] An NVIDIA kernel module 'nvidia-drm' appears to already be loaded in your kernel

[PyTorch] 旅館評論情感分析實戰紀錄 (1)

接續之前旅館評論分類的工作,詳情請參閱《[PyTorch] 旅館評論情感分析實戰紀錄 (0)》這篇文章,我再次對原本的分類模型進行了簡單的改良。至於改良了什麼呢?還記得我在上一篇文章中提到我只是胡亂地將每個字斷開,個別給相異字一個 Index 去代表該字——簡單來講,就只是把文字資料轉成單個數字,好用來進行 Machine Learning。

Read More »[PyTorch] 旅館評論情感分析實戰紀錄 (1)

[PyTorch] 旅館評論情感分析實戰紀錄 (0)

『旅館評論情感分析』是一個適合訓練分類模型的自然語言處理(NLP)任務。今天是我隨意開始嘗試的第一天,於是我找了個 GitHub 上的旅館評論資料集專案,並訓練了一個由全連接層(fully-connected layer)組成的分類模型,並以 accuracy、precision、recall、F1-score 作為我的指標分數評估模型好壞。

Read More »[PyTorch] 旅館評論情感分析實戰紀錄 (0)
Exit mobile version