CLAM复现问题记录

  • 1.按照官网教程配置环境,安装的是cpu版本的torch。需要重新自己安装gpu版本的torch(版本还是一样的,只不过改为gpu的)。建议下载whl后,激活虚拟环境,直接cd到存放whl的目录,然后pip 安装。
  • 2.官网所说的xxxxxxDEVICES=0,在命令行删除掉,这是linux系统所用(我是windows),可以在extract_features_fp.py加入
python 复制代码
print(torch.cuda.current_device())
torch.cuda.set_device(0)
  • 3.extract_features_fp.py脚本中,ctrl+f找到num_workers,改为0。(如果使用笔记本复现测试的话,只有一个卡)
相关推荐
艾上编程17 分钟前
第三章——爬虫工具场景之Python爬虫实战:学术文献摘要爬取,助力科研高效进行
开发语言·爬虫·python
Hi_kenyon27 分钟前
FastAPI+VUE3创建一个项目的步骤模板(二)
python·fastapi
拉普拉斯妖10841 分钟前
DAY38 Dataset和DataLoader
python
Michelle80231 小时前
24大数据 16-1 函数复习
python
dagouaofei1 小时前
AI自动生成PPT工具对比分析,效率差距明显
人工智能·python·powerpoint
ku_code_ku2 小时前
python bert_score使用本地模型的方法
开发语言·python·bert
祁思妙想2 小时前
linux常用命令
开发语言·python
流水落花春去也2 小时前
用yolov8 训练,最后形成训练好的文件。 并且能在后续项目使用
python
Serendipity_Carl2 小时前
数据可视化实战之链家
python·数据可视化·数据清洗
小裴(碎碎念版)2 小时前
文件读写常用操作
开发语言·爬虫·python