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。(如果使用笔记本复现测试的话,只有一个卡)
相关推荐
轮子飞了1 天前
记一次 Spring AI Alibaba + 百炼的踩坑:结构化输出与联网搜索的冲突
人工智能·python·spring
专注VB编程开发20年1 天前
VB.NET是唯一能直接打击 Python 的语言
python
梦想三三1 天前
【NLP入门到实战】TF-IDF算法详解 + 红楼梦120回关键词提取
人工智能·python·计算机视觉
弹简特1 天前
【零基础学Python】05-Python函数完全指南:从初阶定义到进阶参数,一篇打通核心难点
开发语言·python
学地理的小胖砸1 天前
【批量处理tiff文件生成jpg缩略图】
数据库·人工智能·python
机汇五金_1 天前
通信设备防雨箱如何兼顾防护与散热?
网络·python
糖果店的幽灵1 天前
LangChain 1.3 完全教程:从入门到精通-Part 7: Documents(文档处理)
java·python·langchain
Wonderful U1 天前
基于Python爬虫+Django的轻量化天气预报系统:从数据抓取到可视化展示的完整实战
爬虫·python·django
lqjun08271 天前
PyTorch梯度计算
人工智能·pytorch·python
词元Max1 天前
3.1 Agent开发需要懂多少数学?
人工智能·python