一文教你跑通Faster-Rcnn

第一步因为跑这个代码必须使用GPU进行加速,准备用CPU来跑的话,可以放弃了,安装cuda和cudnn,再安装NVIDIA的显卡驱动(版本一定要对),这方面网上教程非常之多,也有不少坑,仔仔细细一步一步的来是不会有多大问题的,安装完一定要进行一个测试,是否安装成功。 测试显卡驱动:$:nvidia-smi,出现GPU的使用情况,则代表成功 测试cuda:$:nvcc -V,出现Cuda的版本信息,
相关文章
相关标签/搜索