linux复现stgcn
时间: 2023-11-07 08:01:47 浏览: 49
复现STGCN的步骤如下:
1. 首先,确保你已经安装了torch。可以使用pip来安装torch,conda install可能会导致与后续安装ffmpeg时出现兼容性问题。
2. 下载并解压与你的CUDA版本对应的cudnn库,并将其中的bin、include、lib文件剪切到CUDA安装目录(一般是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0)下的相应位置。
3. 下载STGCN的源代码,并按照说明进行编译。
4. 准备数据集和模型文件,确保路径的斜杠使用正斜杠(/)。
5. 运行旧版或新版的demo。