PyTorch中apex安裝方式和避免踩坑
背景：這個庫的安裝不是像其他的一樣的直接使用 pip install XXX的形式，而是使用原始的Git方式
1、apex
這是NVIDIA開發(fā)的基于PyTorch的混合精度訓(xù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、雖然就簡單三行命令，但是仍有很多“坑”存在
（1）git clone 克隆失敗————解決方式：直接通過網(wǎng)頁下載
如果出現(xiàn)這種情況，那我們就選擇“曲線救國”。在GitHub中直接下載該庫，然后解壓。并放到相應(yīng)的位置上。
（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的版本不對應(yīng)，但是通過搜索也發(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)文章
這篇文章主要介紹了python__name__原理及用法詳解,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-11-11
這篇文章主要介紹了Django 表單模型選擇框如何使用分組,小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2019-05-05
Python+selenium點擊網(wǎng)頁上指定坐標(biāo)的實例
今天小編就為大家分享一篇Python+selenium點擊網(wǎng)頁上指定坐標(biāo)的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
本文詳細講解了Python并發(fā)編程之IO模型，文中通過示例代碼介紹的非常詳細。對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2022-06-06
Virtualenv 搭建 Py項目運行環(huán)境的教程詳解
這篇文章主要介紹了Virtualenv 搭建 Py項目運行環(huán)境的詳細教程，本文通過圖文并茂的形式給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-06-06
Python+unittest+requests 接口自動化測試框架搭建教程
這篇文章主要介紹了Python+unittest+requests 接口自動化測試框架搭建教程，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-10-10
詳解Python和Rust中內(nèi)存管理機制的實現(xiàn)與對比
Python和Rust都采用了垃圾收集（Garbage?Collection）機制來管理內(nèi)存,但它們各自的實現(xiàn)方式有很大的不同,下面就跟隨小編一起來深入了解下二者的區(qū)別吧2024-03-03
這篇文章主要介紹了python實現(xiàn)rest請求api示例,需要的朋友可以參考下2014-04-04
這篇文章主要給大家介紹了關(guān)于pycharm打開chrome自動退出錯誤的解決方法,文中通過實例代碼以及圖文將解決的辦法介紹的非常詳細,需要的朋友可以參考下2023-07-07
今天小編就為大家分享一篇python 從文件夾抽取圖片另存的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-12-12