uci-cbcl / NoduleNet

[MICCAI' 19] NoduleNet: Decoupled False Positive Reduction for Pulmonary Nodule Detection and Segmentation
Other
185 stars 57 forks source link

build/box error #30

Open jccjd opened 4 years ago

jccjd commented 4 years ago

when i do that

cd build/box
python setup.py install

I got this error

   84 | #pragma omp parallel for if ((end - begin) >= grain_size)
      | 
In file included from box.cpp:4:
nms.cpp: In function ‘int cpu_nms(at::Tensor*, at::Tensor*, at::Tensor*, at::Tensor*, at::Tensor*, float)’:
nms.cpp:7:5: error: ‘AT_CHECK’ was not declared in this scope; did you mean ‘DCHECK’?
    7 |     AT_CHECK(keep_out->is_contiguous(), "argument#1(keep_out) must be contiguous");
      |     ^~~~~~~~
      |     DCHECK
nms.cpp:16:47: warning: ‘T* at::Tensor::data() const [with T = long int]’ is deprecated: Tensor.data<T>() is deprecated. Please use Tensor.data_ptr<T>() instead. [-Wdeprecated-declarations]
   16 |     auto keep_out_flat = keep_out->data<long>();
      |                                               ^
In file included from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Tensor.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Context.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/ATen.h:5,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/types.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader_options.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/base.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/stateful.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/all.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/extension.h:4,
                 from box.cpp:1:
/home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/core/TensorBody.h:354:7: note: declared here
  354 |   T * data() const {
      |       ^~~~
In file included from box.cpp:4:
nms.cpp:17:42: warning: ‘T* at::Tensor::data() const [with T = float]’ is deprecated: Tensor.data<T>() is deprecated. Please use Tensor.data_ptr<T>() instead. [-Wdeprecated-declarations]
   17 |     auto boxes_flat = boxes->data<float>();
      |                                          ^
In file included from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Tensor.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Context.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/ATen.h:5,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/types.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader_options.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/base.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/stateful.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/all.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/extension.h:4,
                 from box.cpp:1:
/home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/core/TensorBody.h:354:7: note: declared here
  354 |   T * data() const {
      |       ^~~~
In file included from box.cpp:4:
nms.cpp:18:41: warning: ‘T* at::Tensor::data() const [with T = long int]’ is deprecated: Tensor.data<T>() is deprecated. Please use Tensor.data_ptr<T>() instead. [-Wdeprecated-declarations]
   18 |     auto order_flat = order->data<long>();
      |                                         ^
In file included from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Tensor.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Context.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/ATen.h:5,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/types.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader_options.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/base.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/stateful.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/all.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/extension.h:4,
                 from box.cpp:1:
/home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/core/TensorBody.h:354:7: note: declared here
  354 |   T * data() const {
      |       ^~~~
In file included from box.cpp:4:
nms.cpp:19:43: warning: ‘T* at::Tensor::data() const [with T = float]’ is deprecated: Tensor.data<T>() is deprecated. Please use Tensor.data_ptr<T>() instead. [-Wdeprecated-declarations]
   19 |     auto areas_flat =  areas->data<float>();
      |                                           ^
In file included from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Tensor.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Context.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/ATen.h:5,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/types.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader_options.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/base.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/stateful.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/all.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/extension.h:4,
                 from box.cpp:1:
/home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/core/TensorBody.h:354:7: note: declared here
  354 |   T * data() const {
      |       ^~~~
In file included from box.cpp:4:
nms.cpp:22:49: warning: ‘T* at::Tensor::data() const [with T = int]’ is deprecated: Tensor.data<T>() is deprecated. Please use Tensor.data_ptr<T>() instead. [-Wdeprecated-declarations]
   22 |     auto suppressed_flat = suppressed.data<int>();
      |                                                 ^
In file included from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Tensor.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Context.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/ATen.h:5,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/types.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader_options.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/base.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/stateful.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/all.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/extension.h:4,
                 from box.cpp:1:
/home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/core/TensorBody.h:354:7: note: declared here
  354 |   T * data() const {
      |       ^~~~
In file included from box.cpp:4:
nms.cpp:96:45: warning: ‘T* at::Tensor::data() const [with T = long int]’ is deprecated: Tensor.data<T>() is deprecated. Please use Tensor.data_ptr<T>() instead. [-Wdeprecated-declarations]
   96 |     auto num_out_flat = num_out->data<long>();
      |                                             ^
In file included from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Tensor.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Context.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/ATen.h:5,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/types.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader_options.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/base.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/stateful.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/all.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/extension.h:4,
                 from box.cpp:1:
/home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/core/TensorBody.h:354:7: note: declared here
  354 |   T * data() const {
      |       ^~~~
In file included from box.cpp:5:
overlap.cpp: In function ‘int cpu_overlap(at::Tensor*, at::Tensor*, at::Tensor*)’:
overlap.cpp:8:5: error: ‘AT_CHECK’ was not declared in this scope; did you mean ‘DCHECK’?
    8 |     AT_CHECK(boxes1->is_contiguous(), "argument#1(boxes1) must be contiguous");
      |     ^~~~~~~~
      |     DCHECK
overlap.cpp:23:44: warning: ‘T* at::Tensor::data() const [with T = float]’ is deprecated: Tensor.data<T>() is deprecated. Please use Tensor.data_ptr<T>() instead. [-Wdeprecated-declarations]
   23 |     auto boxes1_flat = boxes1->data<float>();
      |                                            ^
In file included from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Tensor.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Context.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/ATen.h:5,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/types.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader_options.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/base.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/stateful.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/all.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/extension.h:4,
                 from box.cpp:1:
/home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/core/TensorBody.h:354:7: note: declared here
  354 |   T * data() const {
      |       ^~~~
In file included from box.cpp:5:
overlap.cpp:24:44: warning: ‘T* at::Tensor::data() const [with T = float]’ is deprecated: Tensor.data<T>() is deprecated. Please use Tensor.data_ptr<T>() instead. [-Wdeprecated-declarations]
   24 |     auto boxes2_flat = boxes2->data<float>();
      |                                            ^
In file included from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Tensor.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Context.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/ATen.h:5,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/types.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader_options.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/base.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/stateful.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/all.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/extension.h:4,
                 from box.cpp:1:
/home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/core/TensorBody.h:354:7: note: declared here
  354 |   T * data() const {
      |       ^~~~
In file included from box.cpp:5:
overlap.cpp:25:46: warning: ‘T* at::Tensor::data() const [with T = float]’ is deprecated: Tensor.data<T>() is deprecated. Please use Tensor.data_ptr<T>() instead. [-Wdeprecated-declarations]
   25 |     auto overlap_flat = overlap->data<float>();
      |                                              ^
In file included from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Tensor.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/Context.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/ATen.h:5,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/types.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader_options.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/base.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader/stateful.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data/dataloader.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/data.h:3,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/all.h:4,
                 from /home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/torch/extension.h:4,
                 from box.cpp:1:
/home/mmv2/Downloads/ct_gui/NoduleNet/nodule_env/lib/python3.8/site-packages/torch/include/ATen/core/TensorBody.h:354:7: note: declared here
  354 |   T * data() const {
      |       ^~~~
In file included from box.cpp:5:
overlap.cpp:20:10: warning: unused variable ‘overlap_num’ [-Wunused-variable]
   20 |     long overlap_num = overlap->sizes()[0];
      |          ^~~~~~~~~~~
error: command 'x86_64-linux-gnu-gcc' failed with exit status 1
Min-Sheng commented 3 years ago

I meet the same problem! Does anyone have the solution? @jccjd , did you solve it?

Min-Sheng commented 3 years ago

Hi guys, after I struggle a lot, I solved it! My pytorch vision is 1.5 and CUDA is 10.1 I first install ninja package: pip3 install ninja and replace AT_CHECK in all cpp files with TORCH_CHECK (see the similar issue in mmdetection: https://github.com/open-mmlab/mmdetection/issues/3363)

fengz63 commented 3 years ago

Hi guys, after I struggle a lot, I solved it! My pytorch vision is 1.5 and CUDA is 10.1 I first install ninja package: pip3 install ninja and replace AT_CHECK in all cpp files with TORCH_CHECK (see the similar issue in mmdetection: open-mmlab/mmdetection#3363)

Thanks for your answer. I have solved my question.

adore1979 commented 2 years ago

Hi guys, after I struggle a lot, I solved it! My pytorch vision is 1.5 and CUDA is 10.1 I first install ninja package: pip3 install ninja and replace AT_CHECK in all cpp files with TORCH_CHECK (see the similar issue in mmdetection: open-mmlab/mmdetection#3363)

thank you

frankchen121212 commented 2 years ago

Hi guys, after I struggle a lot, I solved it! My pytorch vision is 1.5 and CUDA is 10.1 I first install ninja package: pip3 install ninja and replace AT_CHECK in all cpp files with TORCH_CHECK (see the similar issue in mmdetection: open-mmlab/mmdetection#3363)

So much thanks!!!! This solution works perfectly!

streaky97 commented 2 years ago

Hi guys, after I struggle a lot, I solved it! My pytorch vision is 1.5 and CUDA is 10.1 I first install ninja package: pip3 install ninja and replace AT_CHECK in all cpp files with TORCH_CHECK (see the similar issue in mmdetection: open-mmlab/mmdetection#3363)

Thank you very much. It helps me a lot!