search

WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
Other
1.68k stars 306 forks source link

Change Maximum allowed Pytorch Version to 10.1 #233

Open manueldiaz96 opened 1 year ago

manueldiaz96 commented 1 year ago

This is a request to specify that the maximum allowed version for Pytorch should be 1.10.1 or 1.10.2.

Motivation After Pytorch 1.11, there were some changes to the CPP code for the THC.h headers to be now part of ATen.h. Knowing this, Pytorch 1.10.1 (no matter if CUDA 10..2 or 11.3), should be the highest allowed Pytorch version unless the CUDA source files for the following source files are updated:

mmdet
├── ops
│     ├── nms/src/
│     │     ├── nms_kernel.cu
│     ├── roi_align/src/
│     │     ├── roi_align_kernel.cu
│     ├── roi_pool/src/
│     │     ├── roi_pool_kernel.cu
│     ├── dcn/src/
│     │     ├── deform_conv_cuda_kernel.cu
│     │     ├── deform_pool_cuda_kernel.cu
│     ├── sigmoid_focal_loss/src/
│     │     ├── sigmoid_focal_loss_cuda.cu
│     ├── masked_conv/src/
│     │     ├── masked_conv2d_kernel.cu