tensorflow를 처음 사용하려고 했을때 버전을 제대로 확인하지 않아서 몇번이나 다시 지우고 깔고를 반복하면서 지치고 시간낭비 햇던 기억이 있었다.
이번엔 한번에 성공하려고 여기저기 찾아보면서 step by step으로 정리하면서 설치해보았다.
내 컴퓨터 환경
윈도우
cuda == 11.2
cudnn == 8.1.0
nvidia driver == 516.94
python == 3.7
pytorch == NVIDIA GeForce GTX 1080 Ti
1. cuda 버전 확인
nvcc -V
# 또는
nvcc --version윈도우 cmd창을 켜서 위의 명령어를 실행한다.
빨간색으로 밑줄 친 부분이 현재 Cuda 버전이다.
2. Cudnn 버전 확인
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\include 경로에 있는 cudnn_version.h 파일을 연다.
cat명령어(리눅스)를 사용하던, 메모장으로 열던 vscode로 열던 이 파일을 열어서 안의 내용을 확인하자.
3. Nvidia 버전 확인
3-1 nvidia-smi 명령어를 실행시켜 nvidia에 대한 대부분의 정보들을 볼 수 있다.
# nvidia driver에 대한 다양한 정보들을 볼 수 있드아.
nvidia-smi실행시킨걸 보면 아래와같은 화면이 뜬다.
1번이 nvidia드라이버 버전이고
2번은 CUDA version이라고 쓰여져있다. 이건 지금 설치되어있는 CUDA버전이 아니다! 사용하고 있는 드라이버의 추천 cuda version이다. 확인해보면 위에서 확인한 버전과 다르다. 헷갈리지 말자.
3번은 현재 gpu사용량을 볼 수 있다. nvidia-smi를 찍었을 그 시점의 gpu사용량을 사진찍듯이 찍은것이라서 현재는 어떤지 모른다. pytorch로 학습하면서 실제로 gpu가 돌아가고 있는지 확인하려면 3번의 숫자를 확인하면 된다.
('nvidia-smi -l 1' 이 명령어를 cmd에 실행시키면 1초마다 nvidia-smi를 찍도록 하면 실시간으로 확인 가능하다. 실제로 해보면 깜빡깜빡인다.)
3-2 nvidia 제어판에서 확인하기.
바탕화면에서 우클릭 -> NVIDIA 제어판 클릭 -> 도움말 탭에서 시스템정보 클릭 -> 드라이버 버전 확인.
아래 스크린샷 참고
이제 cuda, cudnn, nvidia버전을 확인하였으니 이에 맞는 python과 pytorch 가상환경을 만들어 주면 된다.
4. python & pytorch 버전 호환 확인하기.
https://pytorch.org/get-started/locally/
PyTorch
An open source machine learning framework that accelerates the path from research prototyping to production deployment.
pytorch.org
pytorch 공식홈페이지에서 설치하는 명령어를 알려준다. 만약 자기 cuda버전이 없다면 previous pytorch version탭에서 자기한테 맞는 환경을 찾으면 된다.
파이썬 버전은 3.7-3.9까지 지원된다. 파이썬2나 10이상 버전은 안된다.
5. pytorch 가상환경 만들기
우선 conda로 가상환경을 만든다.
# 가상환경 만들기
conda create -n pytorch python=3.7 anaconda -y이름은 'pytorch'로 했고 버전은 3.7로 했다. 뒤에 anaconda옵션을 주면 웬만한 라이브러리는 같이 깔아준다. (나중에 하나하나 깔려면 이것도 귀찮다..) 그리고 -y옵션은 중간에 확인하는 절차가 있는데 그냥 다 yes 해달라는 의미이다.
그리고 cuda 11 버전용을 previous텝에서 찾아서 깔아주었다.
pytorch1.7.1버전을 다운받았다.
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
6. 설치가 잘 되었는지 확인
아래와같이 torch.cuda.is_available() 매서드가 True값이 나오면 설치가 제대로 된 것이다.
>>> torch.cuda.is_available()
True
>>> torch.cuda.current_device()
0
>>> torch.cuda.get_device_name(0)
'NVIDIA GeForce GTX 1080 Ti'
>>> torch.__version__
'1.7.1+cu110'
끝!
읽어주셔서 감사합니다.
피드백 및 지적은 언제나 감사합니다.
좋은하루 되십쇼!
'환경설정' 카테고리의 다른 글
| 가상환경에서 jupyter notebook kernel연결하기 (1) | 2023.01.19 |
|---|