speedinghzl / pytorch-segmentation-toolbox

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

How to use 3090 with CUDA 11? #61

Open chenwang1701 opened 3 years ago

chenwang1701 commented 3 years ago

hi, if I want to use 3090 with CUDA 11, how to change the inplace_abn? some parts of the codes are based on CUDA 9, where to change the code of Inplace-ABN? your original implementation:

Install Pytorch-1.1

$ conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

Install Inplace-ABN

$ git clone https://github.com/mapillary/inplace_abn.git $ cd inplace_abn $ python setup.py install