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。(如果使用笔记本复现测试的话,只有一个卡)
相关推荐
飞Link1 分钟前
K 折交叉验证(K-Fold Cross Validation)全解析:原理、代码实践、应用场景与常见坑点
人工智能·python·机器学习
衫水8 分钟前
如何在离线情况下部署项目(前端VUE + 后端Python)
前端·vue.js·python
【赫兹威客】浩哥10 分钟前
【赫兹威客】框架模板-后端命令行部署教程
python·django
Java后端的Ai之路10 分钟前
【Python小知识】-self是什么?
linux·python··self
Sylvia33.10 分钟前
如何获取足球数据统计数据API
java·前端·python·websocket·数据挖掘
zyb114758243311 分钟前
JVM的学习
jvm·python·学习
Remember_99314 分钟前
文件系统与IO操作:深入解析与Java实践
java·开发语言·数据结构·ide·python·算法
纪伊路上盛名在14 分钟前
Chap2 Neural Networks with PyTorch
人工智能·pytorch·python·深度学习·机器学习
开开心心就好15 分钟前
卸载工具清理残留,检测垃圾颜色标识状态
linux·运维·服务器·python·安全·tornado·1024程序员节
小舞O_o17 分钟前
gitlab文件上传
linux·服务器·git·python·目标检测·机器学习·gitlab