[Ubuntu] pip으로 PyTorch 설치하기 - Linux

start()

{

---

본 블로그는 conda로 pytorch를 설치 하지 않고 pip으로 설치 함을 설명 해 나갑니다.

 

설치 전 확인 사항

일단 pip이 잘 동작 하는지 확인 하기 위해 커맨드창을 열고 아래와 같이 칩니다.

pip3 install

아래와 같이 나오면 잘 동 작 합니다.

Defaulting to user installation because normal site-packages is not writeable
ERROR: You must give at least one requirement to install (see "pip help install")

 

만약 문제가 있다면 제 포스트를 참고 해보시면 될듯 합니다.

https://mickael-k.tistory.com/43

[Ubuntu] pip3 버전 문제 해결하기

---

설치

• 사이트 참고

아래 사이트 참고합니다. 아래 사이트에서 linux환경에서 pip으로 설치 방법을 다루고 있습니다.

https://pytorch.cn/previous-versions/

About PyTorch

설명에 의하면 먼저 아래와 같이 작성을 합니다.

pip3 install http://download.pytorch.org/whl/

 

그 뒤에는 이제 쿠다 버전, 파이썬 버전에 맞춰서 적어야 합니다.

전 쿠다 8.0이며, 파이토치 0.4.0을 설치하고 싶습니다. 또한 파이썬 3.5버전을 갖고 있기에 아래와 같은 명령어를 위 명령어 뒤에 붙여 주어야 합니다.

cu80/torch-0.4.0-cp35-cp35m-linux_x86_64.whl

 

최종적으로 아래와 같이 합니다.

pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-linux_x86_64.whl

 

잘 모르겠으면 위 링크의 파이토치사이트에서 보고 하시면 됩니다.

---

예제 출력

import torch

x = torch.tensor([[1,2,3], [4,5,6], [7,8,9]])
print(x)

 

아래와 같이 나오면 성공

tensor([[ 1,  2,  3],
        [ 4,  5,  6],
        [ 7,  8,  9]])

---

참고

 

https://pytorch.cn/previous-versions/

About PyTorch

---

return 0;

}