pytorch version support
BigBuffa1o opened this issue · comments
BigBuffa1o commented
My GPU is 3090 which run on CUDA11.1,is the code support higher version pytorch?i find problem complie the make.sh
linlin commented
me too
VictorYrotciV commented
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