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)文章
這篇文章主要介紹了python__name__原理及用法詳解,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-11-11
這篇文章主要介紹了Django 表單模型選擇框如何使用分組,小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2019-05-05
Python+selenium點(diǎn)擊網(wǎng)頁(yè)上指定坐標(biāo)的實(shí)例
今天小編就為大家分享一篇Python+selenium點(diǎn)擊網(wǎng)頁(yè)上指定坐標(biāo)的實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-07-07
本文詳細(xì)講解了Python并發(fā)編程之IO模型，文中通過(guò)示例代碼介紹的非常詳細(xì)。對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2022-06-06
Virtualenv 搭建 Py項(xiàng)目運(yùn)行環(huán)境的教程詳解
這篇文章主要介紹了Virtualenv 搭建 Py項(xiàng)目運(yùn)行環(huán)境的詳細(xì)教程，本文通過(guò)圖文并茂的形式給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-06-06
Python+unittest+requests 接口自動(dòng)化測(cè)試框架搭建教程
這篇文章主要介紹了Python+unittest+requests 接口自動(dòng)化測(cè)試框架搭建教程，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-10-10
詳解Python和Rust中內(nèi)存管理機(jī)制的實(shí)現(xiàn)與對(duì)比
Python和Rust都采用了垃圾收集（Garbage?Collection）機(jī)制來(lái)管理內(nèi)存,但它們各自的實(shí)現(xiàn)方式有很大的不同,下面就跟隨小編一起來(lái)深入了解下二者的區(qū)別吧2024-03-03
python實(shí)現(xiàn)rest請(qǐng)求api示例
這篇文章主要介紹了python實(shí)現(xiàn)rest請(qǐng)求api示例,需要的朋友可以參考下2014-04-04
pycharm打開(kāi)chrome自動(dòng)退出錯(cuò)誤的解決方法
這篇文章主要給大家介紹了關(guān)于pycharm打開(kāi)chrome自動(dòng)退出錯(cuò)誤的解決方法,文中通過(guò)實(shí)例代碼以及圖文將解決的辦法介紹的非常詳細(xì),需要的朋友可以參考下2023-07-07
今天小編就為大家分享一篇python 從文件夾抽取圖片另存的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-12-12