1. 程式人生 > >windows下安裝mxnet+GPU

windows下安裝mxnet+GPU

折騰了一整天,windows下的mxnet終於搭建好了。流程說起來很簡單,中間曲折卻不一而足。記錄一下,給大家參考。

首先安裝Anaconda,網上有說mxnet不支援python3,於是我華麗麗地安裝了anaconda3又解除安裝,裝上anaconda2,不信邪,看到mxnet官網沒有說明不能用python3,本著無知者無畏的探索精神,又重新裝了anaconda3.

安裝完anaconda3後,參照文章:

安裝的時候提示:需要更新pip.

我開始按照提示對pip進行更新,更新後還是不能用。

經過排查原因後發現我係統安裝VS2017時安裝的python3與anaconda3衝突了,我更新的是VS2017路徑中安裝的pip,而系統的預設呼叫中的pip為anaconda3中安裝的版本。

最後更改了系統環境變數,去掉了環境變數中,VS2017安裝的python目錄,重新升級pip後,CPU版本和GPU版本均可以安裝。

其中還出現了兩個小細節:

1、開啟cmd進行安裝時,切記要以管理員方式開啟,否則很多命令執行不了;

2、為了便於測試,我安裝mxnet的時候CPU版本和GPU版本都進行了安裝,測試GPU的時候發現GPU用不了。後來將兩個均解除安裝了重新安裝,只裝GPU版本,測試才可以通過。