jwyang / faster-rcnn.pytorch

A faster pytorch implementation of faster r-cnn

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pytorch version support

BigBuffa1o opened this issue · comments

My GPU is 3090 which run on CUDA11.1,is the code support higher version pytorch?i find problem complie the make.sh

commented

me too

pytorch-1.0 branch is ok on rtx 3090, though you may need to make a few modifications on the code according to the error message. My pytorch environment is built by
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html