PyTorch中l(wèi)oading fbgemm.dll異常的解決辦法
一、安裝必備
1. window
學(xué)習或開發(fā)階段，我們通常在window環(huán)境下進行，因此需滿足以下條件：
Windows 7 and greater；
Windows 10 or greater recommended；
Windows Server 2008 r2 and greater；
2. python
準備一個python環(huán)境，需滿足以下條件：
Python 3.8-3.11（支持）；
Python 2.x（不支持）；
3. install pytorch
正式安裝pytorch執(zhí)行如下命令（默認最新版）：
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
安裝successful后，可看到如下界面（來自pycharm）：
二、異常情景
1. 請求GPT-2
博主拿GPT作為示例（來自官方），新建一個python文件并取名為hello_GPT2.py，目的是完成gpt2模型的調(diào)用，下面是源碼 ：
from transformers import GPT2LMHeadModel, GPT2Tokenizer
# 指定模型名稱
model_name = 'gpt2'
# 加載模型和分詞器
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
model = GPT2LMHeadModel.from_pretrained(model_name)
# 輸入文本
input_text = "Once upon a time"
# 對輸入文本進行分詞
inputs = tokenizer.encode(input_text, return_tensors='pt')
# 生成文本
outputs = model.generate(
inputs,
max_length=100, # 生成文本的最大長度
num_return_sequences=1, # 生成序列的數(shù)量
temperature=0.7, # 溫度控制生成的多樣性，值越高，生成的文本越隨機
top_k=50, # 控制生成的詞匯范圍，值越小，生成的文本越集中
top_p=0.9 # 采樣閾值，控制生成的文本多樣性
)
# 解碼生成的文本
generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
print("生成的文本:")
print(generated_text)
2. 異常Traceback
當我們執(zhí)行上面的代碼時，極容易遇到以下異常：
Traceback (most recent call last):
File "D:\projects\PycharmProjects\llm_openai_gpt\hello_GPT2.py", line 1, in <module>
from transformers import GPT2LMHeadModel, GPT2Tokenizer
File "D:\projects\PycharmProjects\llm_openai_gpt\.venv\Lib\site-packages\transformers\__init__.py", line 26, in <module>
from . import dependency_versions_check
File "D:\projects\PycharmProjects\llm_openai_gpt\.venv\Lib\site-packages\transformers\dependency_versions_check.py", line 16, in <module>
from .utils.versions import require_version, require_version_core
File "D:\projects\PycharmProjects\llm_openai_gpt\.venv\Lib\site-packages\transformers\utils\__init__.py", line 34, in <module>
from .generic import (
File "D:\projects\PycharmProjects\llm_openai_gpt\.venv\Lib\site-packages\transformers\utils\generic.py", line 462, in <module>
import torch.utils._pytree as _torch_pytree
File "D:\projects\PycharmProjects\llm_openai_gpt\.venv\Lib\site-packages\torch\__init__.py", line 148, in <module>
raise err
OSError: [WinError 126] 找不到指定的模塊。 Error loading "D:\projects\PycharmProjects\llm_openai_gpt\.venv\Lib\site-packages\torch\lib\fbgemm.dll" or one of its dependencies.
關(guān)鍵之處：
OSError: [WinError 126] 找不到指定的模塊。Error loading "D:\projects\PycharmProjects\llm_openai_gpt\.venv\Lib\site-packages\torch\lib\fbgemm.dll" or one of its dependencies.
三、解決辦法
根據(jù)提示，是因為fbgemm.dll缺少依賴，導(dǎo)致加載異常，所以直接辦法去找依賴文件，博主這里給出一個解決的辦法：
1. 下載libomp140.x86_64.dll文件
點擊 dllme.com后，可看到如下頁面：
點擊右下角，下載 libomp140.x86_64_x86-64.zip。
2. 存放位置
將zip解壓后，有一個文件：libomp140.x86_64.dll，轉(zhuǎn)移至 Windows\System32 目錄下，如存在可覆蓋。
完成后，可順利排除該異常。
結(jié)語
該文用于解決PyTorch2.4安裝后，出現(xiàn)了 OSError: [WinError 126] 找不到指定的模塊，Error loading "PATH\torch\lib\fbgemm.dll" or one of its dependencies.的問題，如存在其他異常，還需進一步探索，如有疑問，歡迎指正！
以上就是PyTorch中l(wèi)oading fbgemm.dll異常的解決辦法的詳細內(nèi)容，更多關(guān)于PyTorch loading fbgemm.dll異常的資料請關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
這篇文章主要介紹了詳解Django緩存處理中Vary頭部的使用,Django是最具人氣的Python web開發(fā)框架,需要的朋友可以參考下2015-07-07
這篇文章主要介紹了python?import模塊時有錯誤紅線的原因及解決，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-02-02
Python 數(shù)據(jù)分析之逐塊讀取文本的實現(xiàn)
這篇文章主要介紹了Python 數(shù)據(jù)分析之逐塊讀取文本的實現(xiàn)，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習或者工作具有一定的參考學(xué)習價值，需要的朋友們下面隨著小編來一起學(xué)習學(xué)習吧2020-12-12
Python自動化辦公之Excel數(shù)據(jù)的寫入
這篇文章主要為大家詳細介紹一下Python中excel的寫入模塊- xlsxwriter，并利用該模塊實現(xiàn)Excel數(shù)據(jù)的寫入，感興趣的小伙伴可以了解一下2022-05-05
這篇文章主要為大家介紹了python異常基類BaseException詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2023-06-06
Python這個編程語言雖然沒有指針類型，但是Python中的可變參量也可以像指針一樣，改變一個數(shù)值之后，所有指向該數(shù)值的可變參量都會隨之而改變，這篇文章主要介紹了Python3中的“指針”,需要的朋友可以參考下2024-02-02
這篇文章主要介紹了python通過pil為png圖片填充上背景顏色的方法,實例分析了Python使用pil模塊操作png圖片的技巧,非常具有實用價值,需要的朋友可以參考下2015-03-03
Python3.12內(nèi)置的71個函數(shù)匯總（按用途分類）
Python 的強大之處在于它自帶了豐富的工具箱,截至 Python 3.12 版本,官方提供了 71 個內(nèi)置函數(shù),這些函數(shù)無需 import 任何模塊即可直接使用,為了方便查閱和記憶,本文將這 71 個函數(shù)按功能分類2026-01-01
分析python動態(tài)規(guī)劃的遞歸、非遞歸實現(xiàn)
本文小編給大家詳細分析了python動態(tài)規(guī)劃的遞歸、非遞歸實現(xiàn)過程以及相關(guān)代碼，有興趣的朋友可以學(xué)習下。2018-03-03
這篇文章主要介紹了Pandas直接讀取sql腳本的方法,本文給大家介紹的非常詳細，對大家的學(xué)習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-01-01