ubuntu18.04复现DenseFusion pytorch1.6.0 LineMOD数据集

还没入门，刚刚跑通了DenseFusion pytorch1.6.0版本的，记录一下踩坑过程
下载源码：DenseFusion pytorch1.6.0
因为按照pytorch1.0失败了，所以选择了1.6.0版本
程序虽然不大，但是数据集很大，有大佬建议建立软链接可以把数据集放在硬盘里，但是我直接把程序放在硬盘里了，就不担心系统内存不够了（主要是没学会软链接）
LineMOD数据集比YCB要小，这两天断网还没有下完YCB所以先用LineMOD数据集试试

环境配置

笔记本显卡应该是2060Ti，太久了记不清具体型号了，反正是20系。
驱动如图：

从base应该能看出是在anaconda下搞的，具体怎么安anaconda就先不提了，直接上主题
创建虚拟环境

conda create -n df-Py1.6 python=3.6

激活虚拟环境：

conda activate df-Py1.6

如果想要退出虚拟环境：

conda deactivate

安装pytorch1.6.0：
我用的pip安装的，因为conda安装失败

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 torchaudio==0.6.0 -f https://download.pytorch.org/whl/torch_st