WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Change Maximum allowed Pytorch Version to 10.1

manueldiaz96 opened this issue · comments

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