cudnn, cudatoolkit을 수동으로 버전 확인하며 까는 건 정말 귀찮다.
아나콘다 가상 환경에 명령어 몇줄로 설치 가능하다.
conda install tensorflow-gpu
이후 conda list 로 cudatoolkit 버전 확인 후 맞는 버전의 torch를 설치한다. 아래의 사이트에서 명령어를 확인한다.
conda list
torch 다운로드