pytorch-segmentation-toolbox icon indicating copy to clipboard operation
pytorch-segmentation-toolbox copied to clipboard

Published 20 hours ago • verified publisher icon speedinghzl

How to use 3090 with CUDA 11?

Open chenwang1701 opened this issue 3 years ago • 0 comments

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

chenwang1701 avatar Mar 11 '21 07:03 chenwang1701