-
[PyTorch] How to run pytorch using conda by linking vscode and jupyter notebook on OSX(mac).AI 2020. 4. 17. 14:26
pytorch로 deeplearning 예제를 로컬(osx)에서 실습해보려고 한다.
에디터는 pycharm을 많이 쓰는거 같던데, 나의 경우 항상 쓰던 vscode를 사용하고자 한다.
예제 소스는 .ipynb 파일로 되어있어서, .py 파일로 변형하지 않고 사용하기 위해 jupyter notebook을 사용하려 한다.
자 그럼 이제 개발 환경을 구축해보자.
1. Install python 3.x 링크
기본적으로 python 3.x를 설치해준다.
(설치 과정은 생략..)
2. Install vscode & extensions.
그리고 확장 프로그램을 설치한다.
설치가 완료되면 vscode를 재시작 해준다.
3. Install jupyter
$ pip3 install --upgrade pip $ pip3 install jupyter
설치가 완료되면 실행해보자
$ jupyter notebook
4. Install anaconda
이제 가장 중요한 단계이다.
pytorch를 사용하려면 가상환경이 필요한데 주로 anaconda를 사용 한다. 링크
패키지 파일을 다운로드해 설치해주자.
(과정은 생략)
설치가 완료 되었다면 pytorch용 가상환경을 만들자.
이때 이름은 자유롭게 지정 할 수 있다.
$ conda create -y -n pytorch ipykernel python=3
가상환경 안으로 들어가자.
$ source activate pytorch_env
이제 pytorch를 설치하자.
(pytorch)$ conda install -y pytorch torchvision -c pytorch
jupyer에 새로운 kernel 등록!
(pytorch)$ python -m ipykernel install --user --name pytorch --display-name "PyTorch"
가상환경 내에서 모듈 설치 명령어 (코드 실행시 모듈이 없다는 에러가 나타날때 사용)
$ conda install -y [모듈명] $ conda install -y ipykernel $ conda install -y seaborn
끝으로 가상환경 빠져나오는 명령어
$ (pytorch)$ source deactivate
5. test
노트북에 new 버튼을 클릭하면 PyTorch가 추가된 것을 확인 할 수 있다.
이제 에러가 발생하지 않는다
vscode에선
우측 상단에 kernel 모드를 pytorch로 변경해주자.
그러면 예제 코드도 이제 잘 동작한다.
참고사이트
https://dreamgonfly.github.io/2018/01/30/conda-pytorch.html
http://taewan.kim/tutorial_manual/dl_pytorch/02/install/mac_env/
'AI' 카테고리의 다른 글
[pytorch] How to use nn.CrossEntropyLoss() 사용법 (0) 2020.04.21 [DeepLearning] 학습 단계를 train, validation, test 로 나누는 이유 (0) 2020.04.20 CNN - pytorch convnet에서 zero padding 값 구하는 공식 (0) 2020.04.16 [tensorflow] How to use contrib.eager.metrics on tensorflow 2.x version (0) 2020.03.25 [tensorflow] How to use tf.contrib.data.make_csv_dataset() on tensorflow 2.x (0) 2020.03.25