caffe使用教程,caffe安装教程windows
1、查看库达和加速库版本
库达版本
cat /usr/local/cuda/version.txt
加速库版本
cat/usr/local/cuda/include/cud nn。h grep cud nn _ MAJOR-A 2
2、可以讲库达和加速库升级到8.0和73、安装咖啡2,用城市安装比较方便,如果环境不匹配可以多试几次
康达安装-c咖啡2咖啡2-cuda8.0-cudnn7
4、之后还要安装NCCL2
安装存储库。对于本地本地重复文件删除辅助工具存储库:sudo dpkg-I nccl-repo-version。黛比
这里首先下载了nccl-repo-Ubuntu 1404-2。1 .15-ga-cuda 8.0 _ 1-1 _ amd64。黛比,之后通过
sudo apt-get install libnccl 2 libnccl-dev安装相应的库,之后运行如下验证代码:#检查咖啡2构建是否成功 python-c from caffe 2 . python import core 2/dev/null echo Success echo Failure #检查Caffe2 GPU构建是否成功#这必须打印一个数字0,以便使用咖啡2。大蟒导入工作空间中的detectron python-c ;打印(工作区 NumCudaDevices())都好的说明咖啡2安装完成。
郑重声明:本文由网友发布,不代表盛行IT的观点,版权归原作者所有,仅为传播更多信息之目的,如有侵权请联系,我们将第一时间修改或删除,多谢。