- 博客(3)
- 收藏
- 关注
原创 裸机到yolov5GPU开始训练
这里写自定义目录标题裸机开始部署显卡驱动+CUDA+cudnnCUDA起手(关键)显卡驱动继续回到CUDAcudnn收尾安装好了环境,到YOLOv5了裸机开始部署显卡驱动+CUDA+cudnnCUDA起手(关键)一般的教程都从安装显卡驱动开始,但是因为现在的各种pytorch版本支持的CUDA不一样,一个小小版本的不合适就会导致最后训练的时候出问题。所以首先我们要确定自己需要的CUDA版本。我们可以使用pytorch官网的配置方法配置,这样出错的概率会比较低,当前网站https://pytorch.
2022-04-08 16:32:22 1502
转载 解决 yolov5训练时CUDNN_STATUS_NOT_INITIALIZED的问题
解决 yolov5训练时CUDNN_STATUS_NOT_INITIALIZED的问题这两天(2021年3月20日左右)在github下载yolov5后准备先使用一下自带的coco128数据集尝试效果,但是每次运行的时候都会报错:RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED查阅了大量的资料,包括重装了CUDA和Cudnn,重装了Anaconda3等等方法,都不行。直到看到了6天前在stackoverflow有人和我出现了同样的问题,报错
2021-03-25 17:12:50 3198 4
原创 Ubuntu18.04+Anaconda3配置Tensorflow-gpu
Ubuntu18.04+Anaconda3配置Tensorflow-gpuUbuntu18.04+Anaconda3配置Tensorflow-gpu1.Anaconda安装配置TensorFlow:其他问题:引用Ubuntu18.04+Anaconda3配置Tensorflow-gpu为了做深度学习训练,在自己的电脑上准备用Anaconda搭建一个环境,记录一下过程和踩过的坑。1.Anaconda安装有些教程先让大家安装CUDA和cuDNN,我也是这样做的,但是发现如果使用conda安装Tensor
2021-03-15 10:17:31 503
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人