PyTorch中apex安裝方式和避免踩坑

Posted on by

背景：這個庫的安裝不是像其他的一樣的直接使用 pip install XXX的形式，而是使用原始的Git方式

1、apex

這是NVIDIA開發的基於PyTorch的混合精度訓練加速神器，能夠增加運算速度，並且減少顯存的占用

2、安裝步驟

git clone https://github.com/NVIDIA/apex.git
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

3、雖然就簡單三行命令，但是仍有很多“坑”存在

（1）git clone 克隆失敗————解決方式：直接通過網頁下載

如果出現這種情況，那我們就選擇“曲線救國”。在GitHub中直接下載該庫，然後解壓。並放到相應的位置上。

（2）安裝步驟中的第三行命令報錯

報錯內容：RuntimeError: Cuda extensions are being compiled with a version of Cuda that does not match the version used to compile Pytorch binaries.  Pytorch binaries were compiled with Cuda 9.0.176.

解決方式：其錯誤意思就是cuda和pytorch的版本不對應，但是通過搜索也發現可以不帶 –global –option 也能用
於是，修改第三行命令為：

pip install -v --no-cache-dir ./

最後的最後，希望大傢都能成功安裝，沖沖沖！

到此這篇關於PyTorch中apex安裝方式和避免踩坑的文章就介紹到這瞭,更多相關PyTorch apex安裝內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！ 

推薦閱讀：