mapillary / inplace_abn

In-Place Activated BatchNorm for Memory-Optimized Training of DNNs
BSD 3-Clause "New" or "Revised" License
1.32k stars 187 forks source link

CUDA 10.1 Failed. Success after update CUDA 11.1 #222

Open PrinceRay7 opened 2 years ago

PrinceRay7 commented 2 years ago

With environment torch==1.8.1 CUDA 10.1 failed the installation. After update CUDA to 11.1 successfully installed. So for those who got problems with installation especially nvcc error, maybe update the cuda version is a good choice, after all you got to update it at some time. Plus, when downloading cuda, always choose runfile instead of deb, it's easier and more convenient and less problems.