3D视觉
里咯胡黑狗
这个作者很懒,什么都没留下…
展开
-
torch.sum,dim参数记录
小白学习pytorch,一直对dim这个参数很迷糊,记录一下 还是小白,说错的地方麻烦指正 假设已经有一个tensor x 了,维度是[2,4,3],2是batch,4是点云个数,3是点云的坐标,即[B,N,C]=[2,4,3] tensor([[[2, 6, 7], [3, 2, 5], [9, 4, 3], [0, 2, 3]], [[0, 6, 6], [8, 2, 8], [0, 9, 4],原创 2020-10-13 17:13:00 · 916 阅读 · 1 评论 -
Ubuntu16.04 安装Kaolin深度学习
Kaolin是什么? 一个用于加速3D深度学习研究的PyTorch库 安装 官方给出的环境要求 1.linux 2.python==3.6 3.CUDA>=10.0.130 要装好nvcc 4.显卡驱动>410.48 5.numpy>=1.17 且 numpy<1.18.4 6. PyTorch >=1.2, <1.5 and Torchvision >=0.4.0 我的安装过程 1.环境: Ubuntu16.04 Driver原创 2020-10-07 15:23:35 · 2272 阅读 · 10 评论