-
Notifications
You must be signed in to change notification settings - Fork 473
遇到......Please exit the torchvision project source and relaunch your python interpreter. segment parameters: 12.4M的解決方法 #159
Copy link
Copy link
Open
Description
各位好,因為我在執行DeepMosaics_0.5.1_gpu勾選GPU後也卡在
......
Please exit the torchvision project source and relaunch your python interpreter.
segment parameters: 12.4M
詢問Gemini後得到以下結論,以及解決方法
Sorry to those who don't speak Mandarin. My English isn't good enough to express what I'm about to say in English. Please use a translator if you need the following instructions.
原因:GPU架構過新導致假死
如果您的顯示卡世代較新(例如RTX30、40或50系列),DeepMosaics內包裝的舊版CUDA檔案可能根本無法正確驅動您的顯示卡,導致在向VRAM 傳輸資料時發生死鎖(Deadlock)而無回應。
驗證方式是:在您原本的執行指令最後面,將參數修改為 --gpu_id -1(這代表強制捨棄GPU,改用CPU運算)。如果改用CPU就能成功越過 segment parameters: 12.4M 並開始輸出圖片,就證實了是軟體內建的GPU模組與你的硬體不相容。
解決方法:放棄免安裝包,使用原始碼並搭配最新版PyTorch
- 到GitHub的DeepMosaics頁面點擊綠色的「Code」圖案,選擇「Downald ZIP」把原始碼下載下來後解壓縮,將檔案放到您喜歡的地方。
- 到Python官網下載並安裝Python。(Gemini推薦我3.10.11版,各位可以試試看不同版本)
- 到NVIDIA官網下載並安裝CUDA Tool Kit。(我選用13.0,不同的版本也是可以的)
- 建立並啟動虛擬環境:到DeepMosaics-master的資料夾內,點選位址列(檔案總管上方平常顯示檔案路徑的那邊),刪除原有的資訊,輸入CMD即可開啟提示命令字元並導航到當前資料夾。開啟提示命令字元後分別輸入以下指令以建立並開啟虛擬環境(每行都是獨立的指令):
===============
python -m venv venv
.\venv\Scripts\activate
===============
注意!虛擬環境啟動成功後,您的命令提示字元最前面會多出一個 (venv) 的標示。接下來的所有安裝與執行,都必須在有這個標示的狀態下進行。 - 執行以下指令以安裝PyTorch與OpenCV(每行都是獨立的指令):
===========================================================
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130
pip install opencv-python
===========================================================
注意!安裝PyTorch指令最後的「cu130」請根據不同CUDA版本號調整,CUDA版本為12.8請改為cu128,以此類推。
另外,您在安裝PyTorch時若出現紅字錯誤訊息,可能是虛擬環境中的pip版本過舊,請先執行以下指令更新pip再重新執行第五步。
=======================
python -m pip install --upgrade pip
======================= - 到原作者的Google Drive下載訓練模型(pretrained_models)解壓縮並放到DeepMosaics-master的資料夾內
恭喜!大功告成
使用方法
- 到DeepMosaics-master的資料夾內,點選位址列(檔案總管上方平常顯示檔案路徑的那邊),刪除原有的資訊,輸入CMD即可開啟提示命令字元並導航到當前資料夾。開啟提示命令字元後輸入以下指令以開啟虛擬環境:
===============
.\venv\Scripts\activate
=============== - 使用以下指令以開始轉檔
===================================================================
python deepmosaic.py --media_path ./檔名.mp4 --model_path ./pretrained_models/模組.pth --gpu_id 0
===================================================================
注意!「./檔名.mp4」代表檔案就在DeepMosaics-master的資料夾內;「./pretrained_models/模組.pth」代表模組檔案在DeepMosaics-master/pretrained_models內。若您的檔案在別處,可以直接替換成對應的「完整路徑」,路徑以/或\分隔都沒關係
原本DeepMosaics_0.5.1_gpu使用的參數(包含 --mode clean、--mode auto、--fps 0、-ss、-t --no_preview等)都是可以加在上列指令中使用的
如有任何問題歡提出討論,謝謝!
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels