caffe使用教程,caffe安装教程windows

  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安装完成。

caffe使用教程,caffe安装教程windows