日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學(xué)無先后,達(dá)者為師

網(wǎng)站首頁 編程語言 正文

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

作者:大夢沖沖沖 ? 更新時(shí)間: 2022-08-27 編程語言

背景:這個(gè)庫的安裝不是像其他的一樣的直接使用 pip install XXX的形式,而是使用原始的Git方式

1、apex

這是NVIDIA開發(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、雖然就簡單三行命令,但是仍有很多“坑”存在

(1)git clone 克隆失敗————解決方式:直接通過網(wǎng)頁下載

如果出現(xiàn)這種情況,那我們就選擇“曲線救國”。在GitHub中直接下載該庫,然后解壓。并放到相應(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),但是通過搜索也發(fā)現(xiàn)可以不帶 --global --option 也能用
于是,修改第三行命令為:

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

最后的最后,希望大家都能成功安裝,沖沖沖!

原文鏈接:https://blog.csdn.net/qq_37844044/article/details/122625442

欄目分類
最近更新