bskyvision RSS 태그 관리 글쓰기 방명록
2020-08-12 15:02:01
728x90

저는 지금까지 Tensorflow를 주로 사용해왔는데, Pytorch도 사용해야 할 일이 생겨서 설치법을 알아보고 설치를 완료했습니다. 

 

CUDA10.0에 맞는 pytorch와 torchvision의 버전은 pytorch 1.2와 torchvision 0.4입니다. 이를 Anaconda에서 설치하려면 다음과 같은 명령어로 설치할 수 있습니다. 

 

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

 

설치가 잘 되었는지 확인하려면, 아래 코드를 실행해보시면 됩니다. 

 

1
2
import torch
torch.cuda.is_available()
cs

 

 

위와 같이 True라고 나오면 제대로 설치된 것입니다. GPU를 이용할 수 있다는 뜻으로 생각하셔도 됩니다. 

 

 

<참고자료>

[1] https://pytorch.org/get-started/previous-versions/, PyTorch, "INSTALLING PREVIOUS VERSIONS OF PYTORCH"

태그 : , ,
댓글

방문해주신 모든 분들을 환영합니다.

* 글을 읽던 중에 궁금했던 부분은 질문해주세요.

* 칭찬, 지적, 의문, 격려, 감사표현 등을 남겨주세요.

* 최대한 답변 드리도록 노력하겠습니다.

* 욕설과 광고를 담은 댓글은 가차없이 삭제합니다.


guest@이름 ~$
guest@패스워드 ~$
guest@홈페이지주소작성 ~$

guest@댓글작성 ~$




bskyvision. Designed by bskyvision.