- 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。(如果使用笔记本复现测试的话,只有一个卡)