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