- 博客(4)
- 收藏
- 关注
原创 DINOv2 基于自定义数据集进行图片分类预测
(前提是安装好pytorch)1. pip安装2. 下载dinov2提供的预训练模型,需要自行前往网站上查找dinov2模型进行下载。模型需要下载三个文件:下载完成后将上述三个文件放置在本地一个名为dinov2_base的文件夹中即可。
2024-03-28 12:57:57
960
原创 anaconda虚拟环境中报错:OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA
然后,对cudnn 进行解压,最后将解压后的 bin,include,lib文件夹下的内容拷贝到 cuda 对应的 bin,include,lib 下即可。本机base环境中没有安装了cuda,也没有配置环境变量,需要从头安装cuda, cuda toolkit和cudnn。在安装mmcv时,anaconda虚拟环境中显示cuda没有安装。又尝试在虚拟环境中采用以下代码,仍然无法解决。ps:这一步我没有做,但是mmcv可以安装了。,和上一步版本匹配的cuda,比如我是。然后按照安装提示进行安装即可。
2024-03-27 10:47:01
5264
8
原创 ModuleNotFoundError: No module named ‘mmcv‘包命令pkgutil报错
第二步:更新setuptools(必须)第三步:根据官方文档选择命令安装。
2024-03-22 16:44:25
637
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人