FGC_grasp复现
复现FGC_grasp
- 环境配置
- 数据集准备
- RuntimeError: CUDA error: invalid device ordinal 问题的解决方案
- raise BadZipFile("File is not a zip file") zipfile.BadZipFile: File is not a zip file问题的解决方案
- 加载数据集时总是被kill然后服务器也卡住了动不了问题的解决方案
环境配置
工程地址: github链接
conda create -n fgcgraspnet python=3.8
conda activate fgcgraspnet
conda install pytorch=1.8.1 torchvision cudatoolkit=11.1 -c pytorch -c nvidia
#测试一下，确保cuda可以使用
python
>>> import torch
>>> torch.cuda.is_available()
True
>>> quit()
pip install ninja #官方文档没说，这里依赖还需要安装ninja库
conda install openblas-devel -c anaconda #安装依赖
conda install cudatoolkit-dev=11.1 -c conda-forge
检查nvcc是否正确：
which nvcc
输出的路径是你conda虚拟环境的路径那就继续下一步(我这里就是自己的虚拟环境里，正确安装的结果)
/home/x822/anaconda3/envs/FGCGrasp/bin/nvcc