본문 바로가기
인터넷/모바일/머신러닝

pycharm에서 torch 인스톨

by blade. 2018. 7. 10.

2019년 8월 7일에 추가.


예전에 쓴 글을 훑다보니, 틀린 것이 있어서 수정함...

예전에는 anaconda 환경 하에서 pycharm IED에서 pytorch를 인스톨하기 위해서 꽤나 번잡스러원 과정을 거쳐야했는데, anaconda가 업그레이드되면서, 언제부터인가 쉽게 인스톨할 수 있게 되었음.


예를 들면 ...


conda install pytorch torchvision cudatoolkit=9.0 -c pytorch


정확한 것은 https://pytorch.org/ 에서 찾아보삼..


###


예전에 썼던 원래 글 시작...



대부분의 module들은 메뉴 -> settings -> Project: -> Project interpreter에서 설치가 가능하지만, 일부 모듈들은 pycharm 내부에서 설치가 불가능. 

torch도 마찬가지..




dos 창을 잠시 띄우고 conda를 이용해서 설치.

먼저 conda 환경을 하나 만들고.. (이미 만들어져있으면, 그냥 쓰면 됨)


1
2
f:\_src\temp>conda create -n pytorch python=3.6
f:\_src\temp>activate pytorch
cs


conda 환경을 실행한 후에, pytorch 홈페이지에서 가져온 설치 명령을 입력한다.


https://pytorch.org/


명령을 입력해보자.


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
 
#############
# 이 명령은 자신의 시스템에 맞게 꿔야함. (pytorch 홈페이지 참고)
#############
(pytorch) f:\_src\temp>conda install pytorch cuda90 -c pytorch 
Solving environment: done
 
## Package Plan ##
  environment location: C:\Users\blade\.conda\envs\pytorch
  added / updated specs:
    - cuda90
    - pytorch
 
The following packages will be downloaded:
 
    package                    |            build
    ---------------------------|-----------------
    numpy-1.14.5               |   py36h9fa60d3_3          94 KB
    mkl_fft-1.0.1              |   py36h452e1ab_0         113 KB
    intel-openmp-2018.0.3      |                0         1.7 MB
    numpy-base-1.14.5          |   py36h5c71026_3         3.8 MB
    cffi-1.11.5                |   py36h945400d_0         213 KB
    mkl_random-1.0.1           |   py36h9258bd6_0         267 KB
    mkl-2018.0.3               |                1       178.1 MB
    blas-1.0                   |              mkl           6 KB
    ------------------------------------------------------------
                                           Total:       184.3 MB
 
The following NEW packages will be INSTALLED:
    blas:         1.0-mkl
    cuda90:       1.0-0                              pytorch
    pytorch:      0.4.0-py36_cuda90_cudnn7he774522_1 pytorch [cuda90]
    그 외 이것저것 같이 설치됨.
 
Proceed ([y]/n)? y
 
Executing transaction: done
 
(pytorch) f:\_src\temp>pip3 install torchvision
Requirement already satisfied: torchvision in c:\users\blade\anaconda3\lib\site-packages (0.2.1)
Requirement already satisfied: numpy in c:\users\blade\anaconda3\lib\site-packages (from torchvision) (1.14.0)
Requirement already satisfied: pillow>=4.1.1 in c:\users\blade\anaconda3\lib\site-packages (from torchvision) (5.0.0)
Requirement already satisfied: six in c:\users\blade\anaconda3\lib\site-packages (from torchvision) (1.11.0)
Requirement already satisfied: torch in c:\users\blade\anaconda3\lib\site-packages (from torchvision) (0.4.0)
 
(pytorch) f:\_src\temp>
cs


pycharm을 다시 띄우면 torch가 등록되어있음.