pytorch 1.7.1 + cuda 11.0 + cudnn 8.0.5.39 on windows

rtx 30xx번대 ampere 아키텍처 정식 지원 pytorch 1.7.1(stable) 버전이 나왔다.

 

기존 1.7.0 버전에서는 30xx번대에서 jit 오류를 뿜어댔으나

 

나를 포함한 다른 유저들의 열화와 같은 성원에

 

nightly 버전(1.8)부터 해결되더니 드디어 stable(1.7.1) 버전도 나왔다.

 

아마 기존에는 anaconda의 cuda 버전이 낮아서 안되지 않았나 싶다.

 

각설하고, 윈도우 환경에서 pytorch 1.7.1 + cuda 11.0 + cudnn 8.0.5로 기존 코드들 깔끔하게 동작한다.

 

30xx번대 그래픽 카드 사용자들은 pytorch 업데이트 해서 사용해보자.