整体配置方式安装官方给的方式即可,但是会存在一些问题,如库冲突等。
- Python 3.7
- Pytorch3d 0.5.0(安装完requirements后自行安装,还需安装ffmpeg=4.0=hcdf2ecd_0,还需更改torch版本,pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html )
- process data阶段的TensorFlow要求pip install tensorflow-gpu==2.6.0.如果TensorFlow变成2.6.0后还是报错,多半是安装TensorFlow时把torch版本替换了,再次用上面的pip install一下就好了
- run阶段用tensorflow-gpu==2.6.0貌似好像也能行,错了,虽然2.6在run时能成功,但是tensorboard启动不了,所以还是乖乖在run时用tensorflow-gpu==1.15.2吧
- 处理256*256的视频时,最好把process的01 2 63457分开跑,否则跑第二步时可能会报cuda out of memory