1-2(Extra). 딥러닝 개발환경 구축(PyTorch 설치)

안녕하세요. 오랜 기간동안 포스팅을 하지못하고, 이제서야 포스팅을 합니다.

군대에 다녀오느라 처리할 시간이 너무 부족했었습니다.

 

---

각설하고, 오늘은 저번의 포스팅인 TensorFlow가 아닌 PyTorch라는 패키지를 설치해보겠습니다.

 

PyTorch란 무엇인가?

PyTorch는 페이스북에서 만든 딥러닝 프레임워크입니다. 

TensorFlow는 구글에서 만든 딥러닝 프레임워크입니다.

 

자세한 내용은 다른 아티클을 참고하시면 좋을거 같고, 이 정도로만 알면될것같습니다.

 

PyTorch는 자율주행전기자동차로 유명한 Tesla 사에서도 사용하고 있고, 여러 논문에서도 입지가 넓어지고 있습니다.

 

그렇기 때문에, Study용으로 이제 올릴 포스팅에는 PyTorch로 사용해보도록 하겠습니다.

---

간단한 conda 커맨드는 저번시간에 설명했으므로, 이번 수업은 빠르게 진행해보겠습니다.

 

※ 혹시 다른 가상환경에 만들고자 한다면, 저번 수업을 참고해 가상환경을 만들어 설치하시면 좋을 것 같습니다!

 

---

PyTorch를 설치하기 전에, 공부할 때 사용할 주요 패키지로는

• Torch : 주 사용하게 될 패키지
• Torchvision : 유명한 데이터셋이나, 모델, computer vision을 위한 패키지

가 있습니다.

 

먼저 Torch를 설치 하겠습니다.

conda install pytorch 커맨드를 통해서 pytorch 설치를 진행합니다.

당연히 y | n 이 나오면 y를 입력해주셔서 넘어가면 됩니다.

 

 

그 다음은 torchvision을 설치해야 하는데, conda install torchvision 커맨드를 입력했는데, 찾을 수 없다고 나옵니다.

그래서 PyTorch에서 Torchvision 공식 문서를 들어가보니, conda 인스톨 시에는

 

conda install torchvision -c pytorch

 

를 이용해서 설치하라고 되어있습니다. 아마 conda의 default 채널에서는 제공하지않아, pytorch 커스텀 채널에서 받아오는 것 같습니다.

 

 

똑같이 뒤는 진행해주면됩니다.

 

그리고 나서 

conda list 를 통해 설치가 잘 되었는지 확인해봅니다.

네, 설치가 잘 된걸 볼 수 있습니다.

 

그리고나서 이제

python 을 입력하셔서, python 쉘을 실행시켜봅시다.

 

import torch

import torchvision

커맨드를 이용해서, 라이브러리를 추가해봅시다.

torch는 정상적으로 추가가 되는데, torchvision을 임포트할 때 웬걸 갑자기!

이런 오류가 막 뜨기시작합니다.

아래 쪽을 보니,

File "C: ~~~~~~~ \torchvision\transforms\functional.py" , line 5

    from PIL import Image, ImageOps, ImageEnhace, PILLOW_VERSION

ImportError : cannot import name 'PILLOW_VERSION' from 'PIL' 이라는 문구가 나옵니다.

엥, 건들지도않은 PIL이라는 곳에서 오류가 납니다.

PIL은 Python Image Library 로 파이썬에서 이미지처리를 도와주는 패키지입니다.

 

오류를 보아 PILLOW_VERSION 이 PIL패키지에서 추가할 수 없다는 내용 같은데,

한번 뒤의 

C:~~~~\Anaconda3\lib\site-packages\PIL\__init__.py

로 들어가봅시다.

메모장으로 그냥 열으셔도 되지만, 가독성이 정말 불편하므로, Pycharm으로  열어줍시다.

 

11번째 라인과, 18~20번째 라인을 보니

PILLOW_VERSION 은 7.0.0버젼에서 삭제되었다고 나오네요

그 대신 PIL.__versio__을 사용하라고 합니다.

 

그리고 저 위의 anaconda prompt에 뜬 오류를 보면

 

    from PIL import Image, ImageOps, ImageEnhace, PILLOW_VERSION 이 오류가 뜬 파일이

C:~~~~~\torchvision\transforms\functional.py 입니다. 

이 것 또한 Pycharm으로 열어봅시다.

 

line 5 에서 오류가 났다고 하니, 5번째 코드를 주목해봅시다.

 

어? PILLOW_VERSION을 사용하지말라고 했는데, 버젓이 사용되고 있습니다.

지워줍시다.

 

그러면 아래 코드에서 PILLOW_VERSION을 불러온 곳이 오류가 날 수 있으니, 어느 부분에 있는지 찾아봅시다.

ctrl + f 로 PILLOW_VERSION을 검색해주세요.

그러면 딱 한군데에서 PILLOW_VERSION이 검색됩니다.

사용하지말랬는데, 왜 사용되고 있는지 잘 모르겠습니다.

 

PIL 코드에서 권고해준 PIL.__version__으로 바꿔줍시다.

저장해줍시다.

 

그리고 아까처럼 다시 python 쉘로 들어간 뒤

import torchvision을 해봅시다.

 

네, 아무것도 뜨지않았으면, import가 잘 된것입니다.

이렇게 되면, 필요한 패키지인 torch, torchvision을 잘 설치했습니다.

 

이 오류해결법이 맞을지는 모르겠지만, 혹여나 진행을 하다가 오류가 나면 다시 정확히 찾아봐야겠습니다.

 

여기까지 Pytorch 설치를 마치겠습니다.

이제 그 다음부터 진짜로 시작해봅시다!