Ubuntu(GPU) 및 MacOS(CPU)가 Conda 샌드박스 환경에 PyTorch 1.0.0 설치

샌드박스 환경의 장점은 그것이 독립적이어서 다른 환경에 영향을 주지 않는다는 데 있다.PyTorch 1.0.0은 Mask R-CNN 및 기타 오픈소스 프로젝트를 쉽게 마이그레이션할 수 있도록 페이스북이 발표한 프레임워크입니다.GPU를 설치하면 먼저 기계의 GPU가 설정되어 있고 튜토리얼 링크가 완료되었음을 보증해야 합니다.참고: Ubuntu 16.04 + CUDA 9.0 + CUDNN 7.1.4를 사용하고 CUPU는 MacOS 10.14.1을 기반으로 합니다.
1 Pytorch 1.0.0 환경 만들기
주의: 여기는python 3.6.5의 환경을 설치하여 자신의 요구에 따라 수정할 수 있습니다

conda create --name pytorch-1.0.0 python=3.6.5

2 환경 관리를 위한 기본 명령
2.1 환경 활성화

source activate pytorch-1.0.0

2.2 현재 활성화된 환경 닫기

source deactivate

2.3 환경 제거

conda remove --name pytorch-1.0.0 --all

2.4 현재 환경 보기

conda info --envs

2.5 이름 변경
clone에서 이름을 바꿔야 하는 환경을 삭제합니다.

conda create --name new_name --clone old_name
conda remove --name old_name --all

3 PyTorch 1.0.0 설치
PyTorch에 들어가서 Preview을 클릭하고 OS, Package, Language, CUDA을 선택하여 페이지에서 제시한 명령을 복사합니다.현재 환경을 활성화하고 설치합니다.
3.1 Ubuntu 16.04

source activate pytorch-1.0.0
conda install pytorch torchvision -c pytorch

3.2 MacOS 10.14.1

source activate pytorch-1.0.0
conda install pytorch torchvision -c pytorch

간단하게 검증을 해볼게요.

python -c "import caffe2; import torch; import torchvision"

4 추가 라이브러리 설치
다음 두 가지 방법은 conda와 pip가 모두 적용되며 취향에 따라 선택하면 됩니다.Opencv,jupyter, tqdm를 설치하는 것을 예로 들겠습니다.먼저 활성화하고 설치하는 방법을 추천합니다
4.1 베이스 환경에서 직접 환경을 설정하여 설치

conda install --name pytorch-1.0.0 opencv jupyter
pip install -t /home/czy/miniconda3/envs/pytorch-1.0.0/lib/python3.6/site-packages/ tqdm

4.2 환경 활성화 후 설치

source activate pytorch-1.0.0
conda install opencv jupyter
pip install tqdm

5 Notebook 및 PyCharm 사용 샌드박스 환경
5.1 Jupyter Notebook
현재 환경을 활성화한 다음 Notebook을 엽니다.

source activate pytorch-1.0.0
jupyter notebook

5.2 PyCharm
프로젝트를 불러올 때, 이 프로젝트의 컴파일 환경을 현재 샌드박스 환경으로 선택하면 됩니다.
Reference
Conda 홈페이지 PyTorch 홈페이지

이 내용에 흥미가 있습니까?

현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:

[Caffe] mnist 인식 프로세스

cd $CAFFE_ROOT 트레이닝 데이터 다운로드 ./data/mnist/get_mnist.sh 데이터 세트 만들기: ./examples/mnist/create_mnist.sh 트레이닝 모델: ./examples/...

텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.

CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.

Ubuntu(GPU) 및 MacOS(CPU)가 Conda 샌드박스 환경에 PyTorch 1.0.0 설치

좋은 웹페이지 즐겨찾기