pytorch install 기록
- window에 CUDA와 CuDNN 설정되어 있다고 가정.
설치
window의 CUDA 버전확인
window CMD 창에서 입력.
nvcc --version명령어를 입력하면 9점대 버전인 것을 확인할 수 있다.
해당 버전에 맞는 Pytorch 설치
pytorch는 anaconda에서 설치하는 것이 편하다.
anaconda prompt 창에서 입력. (설치하고 싶은 가상환경 activate후 입력)
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
설치 확인
torch를 불러오고 device_name을 확인한다.
import torch |
torch.cuda.get_device_name(0) |
