@[TOC] linux下 Pointnet pytorh 版本
NVIDIA-SMI 460.56 Driver Version: 460.56 CUDA Version: 11.2 复现时产生的问题以及解决方法
conda install pytorch=1.7.1 cudatoolkit=11.0 -c pytorch
https://www.pythonheidong.com/blog/article/182191/ccb0fb20d587dc92648a/
https://www.pythonheidong.com/blog/article/182191/ccb0fb20d587dc92648a/
python show_cls.py --model cls/cls_model_0.pth
运行show_cls.py
python show_cls.py --model cls/cls_model_0.pth
运行show_seg.py
python show_seg.py --model seg/seg_model_Chair_3.pth --class_choice Airplane
查看已经安装的Pytorch对应的CUDA版本
进入python环境,输入
import torch
torch.version.cuda
查看已经安装的Pytorch对应的CUDNN版本
进入python环境,输入
import torch
torch.backends.cudnn.version()
#查看是否是gpu版本
flag = torch.cuda.is_available()
print(flag)
训练 train_classification.py
python train_classification.py --dataset ‘/home/wnang/pointnet.pytorch/shapenetcore_partanno_segmentation_benchmark_v0’ --nepoch=‘5’ --dataset_type ‘shapenet’
训练 train_segmentation.py
python train_segmentation.py --dataset ‘/home/wnang/pointnet.pytorch/shapenetcore_partanno_segmentation_benchmark_v0’ --nepoch=‘5’
https://www.freesion.com/article/1470918699/
(pointnet) wnang@wnang-OMEN-Laptop-15-ek0xxx:~/pointnet.pytorch/utils$ python show3d_balls.py
pip install opencv-python
no mudule named pointnet 解决方法: 在segmentation中加入以下字段
import sys
sys.path.append("…/")
顺序可能有点问题 ,为了方便自己看的, 先这样了咯,拜了个拜~