專案地址：PF-AFN
上午de了兩個小時的bug，終于把模型跑通了！
官方的result：
我機器上的result：
哈哈哈哈哈！
記錄下自己從github上下載模型后開始到能運行模型的整個debug程序，
Debug之路
先是搭建模型運行環境，主要是GPU-PyTorch和CUDA，安裝GPU-PyTorch花了很長時間，后來在某寶上花了30RMB請人幫忙裝了一個😂，
從CUDA官網安裝CUDA速度實在太慢，后來就直接在網上搜了網友已經下載好的CUDA安裝包，于是CUDA的安裝問題就順利解決了，感謝熱心網友！有必要提醒一句的是，CUDA的版本一定要和自己電腦的GPU版本適配，
這個模型需要用到cupy，下載與自己電腦上CUDA版本適配的cupy版本即可，
然后就是跑模型了，我是在PyCharm中打開這個專案的，嘗試過在終端運行python test.py 結果不行，老是GPU空間不足的例外，后來直接復制專案里自帶的test.sh中的命令，然后報了有未預料到的XX執行緒存在的例外，后來了解到是開了多執行緒，于是通過改代碼將多執行緒關掉，
再次跑測驗后，還是會報GPU空間不足的例外，在查資料后，將test.py中的
gen_outputs = gen_model(gen_inputs)
改為
with torch.no_grad():
gen_outputs = gen_model(gen_inputs)
模型終于成功跑通！
轉載請註明出處，本文鏈接：https://www.uj5u.com/qukuanlian/271618.html
標籤：區塊鏈