caffe学习笔记——cuda显卡以及驱动问题

运行sh训练文件时,cuda报错

Check failed:error == cudaSuccess(38 vs. 0)

查网上资料怀疑是显卡驱动未成功安装(但前一天还能够成功训练)。因此按照以下步骤进行查看。

一.查看ubuntu下显卡驱动是否安装成功

首先得安装

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
首先,在 Ubuntu 20 上安装 Caffe-CUDA 需要满足一些先决条件。请确保你已经安装了 CUDA 和 cuDNN,并且已经正确配置了它们。然后,按照以下步骤进行安装: 1. 安装依赖项: ```shell sudo apt-get update sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler libgflags-dev libgoogle-glog-dev liblmdb-dev ``` 2. 克隆 Caffe 仓库: ```shell git clone https://github.com/BVLC/caffe.git ``` 3. 进入 Caffe 目录: ```shell cd caffe ``` 4. 编辑 Makefile.config 文件: ```shell cp Makefile.config.example Makefile.config nano Makefile.config ``` 确保按照你的配置修改以下行: - `USE_CUDNN := 1` (如果已安装 cuDNN) - `CUDA_DIR := /usr/local/cuda` (CUDA 的安装路径) - `INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial` (根据你的系统配置修改) - `LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial` (根据你的系统配置修改) 5. 编译 Caffe: ```shell make all -j$(nproc) # 使用所有可用的 CPU 内核 make test -j$(nproc) make runtest -j$(nproc) ``` 6. 安装 Caffe: ```shell sudo make install ``` 7. 配置环境变量: ```shell echo 'export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH' >> ~/.bashrc source ~/.bashrc ``` 现在,你应该已经成功在 Ubuntu 20 上安装了 Caffe-CUDA。你可以尝试运行一些示例代码来验证安装是否成功。记得替换 `/path/to/caffe` 为你的 Caffe 目录路径。希望对你有所帮助!

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值