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。(如果使用笔记本复现测试的话,只有一个卡)
相关推荐
烤汉堡18 分钟前
Python入门到实战:网络请求与数据获取
python
rimoyee21 分钟前
[python探本] 内存数据存储机制
python
LiLiYuan.23 分钟前
Arrays类和List接口的关联
java·开发语言·windows·python
AI小云26 分钟前
【Numpy数据运算】Numpy速学手册
python·numpy
咕白m6253 小时前
通过 Python 在 PDF 中添加页面
python
wa的一声哭了3 小时前
Linux服务器配置ssh免密登陆多台服务器、服务器别名配置
linux·运维·服务器·网络·arm开发·python·ssh
咖啡の猫4 小时前
Python中的输出函数
开发语言·数据库·python
朱一头zcy4 小时前
下载安装pycharm 并通过pycahrm来驱动Anaconda来运行Python程序
ide·python·pycharm
CodeLongBear5 小时前
Python数据分析 -- Pandas基础入门学习笔记:从核心概念到实操代码
python·conda·pandas
B站_计算机毕业设计之家5 小时前
python手写数字识别计分系统+CNN模型+YOLOv5模型 深度学习 计算机毕业设计(建议收藏)✅
python·深度学习·yolo·计算机视觉·数据分析·cnn