一文教你跑通Faster-Rcnn

第一步因爲跑這個代碼必須使用GPU進行加速,準備用CPU來跑的話,可以放棄了,安裝cuda和cudnn,再安裝NVIDIA的顯卡驅動(版本一定要對),這方面網上教程非常之多,也有不少坑,仔仔細細一步一步的來是不會有多大問題的,安裝完一定要進行一個測試,是否安裝成功。 測試顯卡驅動:$:nvidia-smi,出現GPU的使用情況,則代表成功 測試cuda:$:nvcc -V,出現Cuda的版本信息,
相關文章
相關標籤/搜索