speedinghzl / pytorch-segmentation-toolbox

PyTorch Implementations for DeeplabV3 and PSPNet
MIT License
772 stars 167 forks source link

About cuda version. #48

Open chccgiven opened 5 years ago

chccgiven commented 5 years ago

Hi, Thanks to provide the newer code of pytorch 1.1, but my GPU is RTX 2070 and it don't support cuda 9.0, I want to know if this code can run in cuda 10.0? Thank you!

speedinghzl commented 5 years ago

Yes, It could run under Cuda 10.0.