mmfewshot
mmfewshot copied to clipboard
chamfer_distance_forward miss in module _ext
python ./tools/classification/train.py configs/classification/baseline/cub/baseline_conv4_1xb64_cub_5way-1shot.py
No CUDA runtime is found, using CUDA_HOME=':/usr/local/cuda-11.1'
Traceback (most recent call last):
File "./tools/classification/train.py", line 13, in
File "/anaconda3/envs/openmmlab/lib/python3.7/site-packages/mmcv/ops/init.py", line 10, in
my cuda is 11.1,ubuntu 18.04 ,sth is wrong?
torch is 1.12