저는 지금까지 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"
'Dev > python' 카테고리의 다른 글
| [PyTorch] 파이토치 텐서의 최대값 및 그것의 인덱스 (0) | 2020.08.20 |
|---|---|
| [PyTorch] 파이토치에서 행렬 곱셈과 element-wise 곱셈 (0) | 2020.08.20 |
| [PyTorch] 파이토치로 n차원 텐서(행렬) 생성하기 (0) | 2020.08.19 |
| [python] ImportError: cannot import name 'PILLOW_VERSION' from 'PIL' 해결 방법 (0) | 2020.08.12 |
| [python] 현재 PC에 설치되어 있는 CUDA 버전 확인하기 (0) | 2020.08.12 |
| [python] DISTS 파이썬 코드 실행하기(IQA_pytorch 패키지) (4) | 2020.08.10 |
| [python] SSIM 파이썬 코드 실행하기(IQA_pytorch 패키지) (0) | 2020.08.10 |
| [python] PCA를 이용해서 128차원 특성을 10차원으로 축소하기 (0) | 2020.07.31 |