다음에 또 설치할수도 있으니깐 적어놓는 내 노트
(1). 원하는 버전을 확인한다.(토치)
나같은 경우에는 nlp라이브러리를 사용하기 위하여 1.5 이상 1.6 미만의 버전이 필요했다.
그리고 토치, 텐서, 케라스 등등을 사용할때는 gpu가 필수이므로 토치비전과 함께 콘다에 설치해주는것이 좋다.
(2) 쿠다 버전확인 하기
cmd창을 열어서 cuda버전을 확인한다.
nvcc –version
결과는 10.2였다.
그래서 토치비전 1.5.1과 함께 설치 하였다. (아래의 링크로 들어가면 있음)
pytorch.org/get-started/previous-versions/
# CUDA 10.2
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.2 -c pytorch
끝
