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。(如果使用笔记本复现测试的话,只有一个卡)
相关推荐
站大爷IP4 分钟前
Python文件/目录比较实战:排除特定类型的实用技巧
python
Yn31219 分钟前
如何安装 scikit-learn Python 库
python·机器学习·scikit-learn
芥末章宇29 分钟前
Jetson NX Python环境搭建:使用APT轻松安装NumPy, scikit-learn, OpenCV
python·numpy·scikit-learn
猫头虎2 小时前
用 Python 写你的第一个爬虫:小白也能轻松搞定数据抓取(超详细包含最新所有Python爬虫库的教程)
爬虫·python·opencv·scrapy·beautifulsoup·numpy·scipy
三年呀2 小时前
**超融合架构中的发散创新:探索现代编程语言的挑战与机遇**一、引言随着数字化时代的快速发展,超融合架构已成为IT领域的一种重要趋势
python·架构
Q_Q19632884752 小时前
python基于Hadoop的超市数据分析系统
开发语言·hadoop·spring boot·python·django·flask·node.js
MediaTea3 小时前
Python 第三方库:Requests(HTTP 客户端)
开发语言·网络·python·网络协议·http
AI大法师3 小时前
Python:PyQt5 全栈开发教程,构建跨平台桌面应用
python·pyqt
华科云商xiao徐3 小时前
分布式爬虫双核引擎:Java大脑+Python触手的完美协同
java·爬虫·python