[Pytorch] 1.5.0 버전 설치하기

2020. 5. 5. 20:34분석 Python/Pytorch

728x90

(20.05.05 기준)

pytorch에서 mixture_same_family를 사용하고 싶은데, 1.5.0 버전에서만 지원이 되는 것을 확인하였다.

그래서 찾다보니 아래 사이트에서 알려주는 방법으로 일단 설치를 하려고 하니 자꾸 1.4.0이 설치가 되려고 한다.

conda install pytorch torchvision cpuonly -c pytorch

https://pytorch.org/

 

PyTorch

An open source deep learning platform that provides a seamless path from research prototyping to production deployment.

pytorch.org

그래서 필자는 anaconda 환경을 쓰고 있고 anaconda의 환경 이름도 pytorch라고 해놓은 상태이고, cuda 10.0이 설치가 되어 있어서 gpu버전을 쓸 수가 없어서 일단은 cpu 버전으로 설치했다.

 

pytorch 1.5.0 패키지 설치가 위와 같은 방법으로 설치가 안될 시 아래와 같은 방법을 사용해서 설치하면 될 것 같다.

conda install pytorch==1.5.0 torchvision==0.6.0 cpuonly -c pytorch

 

 

 

1.2.0 버전 이하로 설치하는 방법들 소개해줌

https://pytorch.org/get-started/previous-versions/

 

PyTorch

An open source deep learning platform that provides a seamless path from research prototyping to production deployment.

pytorch.org

 

728x90