PyTorch中apex安裝方式和避免踩坑
背景：這個庫的安裝不是像其他的一樣的直接使用 pip install XXX的形式，而是使用原始的Git方式
1、apex
這是NVIDIA開發(fā)的基于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 克隆失敗————解決方式：直接通過網(wǎng)頁下載
如果出現(xiàn)這種情況，那我們就選擇“曲線救國”。在GitHub中直接下載該庫，然后解壓。并放到相應的位置上。
（2）安裝步驟中的第三行命令報錯
報錯內(nèi)容：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的版本不對應，但是通過搜索也發(fā)現(xiàn)可以不帶 --global --option 也能用
于是，修改第三行命令為：
pip install -v --no-cache-dir ./
最后的最后，希望大家都能成功安裝，沖沖沖！
到此這篇關(guān)于PyTorch中apex安裝方式和避免踩坑的文章就介紹到這了,更多相關(guān)PyTorch apex安裝內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了python3中dict（字典）的使用方法，文中給出了詳細的功能列舉，對大家具有一定的參考價值，需要的朋友們下面來一起看看吧。2017-03-03
2021年值得向Python開發(fā)者推薦的VS Code擴展插件
這篇文章主要介紹了2021年值得向Python開發(fā)者推薦的VS Code擴展插件，幫助大家更好的利用vscode進行python的開發(fā)，感興趣的朋友可以了解下2021-01-01
機器人自動回復在很多場景中都可以用的上，本文主要介紹了python制作機器人的實現(xiàn)方法，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-06-06
Python使用oslo.vmware管理ESXI虛擬機的示例參考
oslo.vmware是OpenStack通用框架中的一部分，主要用于實現(xiàn)對虛擬機的管理任務，借助oslo.vmware模塊我們可以管理Vmware ESXI集群環(huán)境。2021-06-06
python實現(xiàn)不同文件夾下的函數(shù)相互調(diào)用
這篇文章主要介紹了python實現(xiàn)不同文件夾下的函數(shù)相互調(diào)用方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08
Python中的并發(fā)處理之a(chǎn)syncio包使用的詳解
本篇文章主要介紹了Python中的并發(fā)處理之a(chǎn)syncio包使用的詳解，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-04-04
這篇文章主要介紹了深入解析Python中的lambda表達式的用法,包括其與def之間的區(qū)別,需要的朋友可以參考下2015-08-08
python opencv實現(xiàn)直線檢測并測出傾斜角度(附源碼+注釋)
這篇文章主要介紹了python opencv實現(xiàn)直線檢測并測出傾斜角度(附源碼+注釋)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-12-12
這篇文章主要介紹了Pandas之ReIndex重新索引的實現(xiàn)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-06-06
這篇文章主要介紹了python使用 __init__初始化操作,結(jié)合實例形式分析了Python面向?qū)ο蟪绦蛟O(shè)計中使用__init__進行初始化操作相關(guān)技巧與注意事項,需要的朋友可以參考下2019-09-09