1. 아나콘다 설치

https://www.anaconda.com/products/individual

 

Anaconda | Individual Edition

Anaconda's open-source Individual Edition is the easiest way to perform Python/R data science and machine learning on a single machine.

www.anaconda.com

 

2. 텐서플로우, Pytorch 버전 호환 확인

https://www.tensorflow.org/install/source

 

소스에서 빌드  |  TensorFlow

Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge 소스에서 빌드 소스에서 TensorFlow pip 패키지를 빌드하고 Ubuntu Linux 및 macOS에 설치합니다. 명령어는 다른 시스템에도 적용될 수 있지만, U

www.tensorflow.org

 

 

https://pytorch.org/

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

 

PyTorch

An open source machine learning framework that accelerates the path from research prototyping to production deployment.

pytorch.org

 

3. 아나콘다 가상환경 생성 뒤, 아래 예시와 같이 설치하면 됨

<텐서플로우>

(case 1)
conda update conda
conda update --all
conda create -n tensorflow_env python=3.8
conda env list
activate tensorflow
conda install tensorflow-gpu=버전

(case 2) : (case 1)이 먹통인 경우, RTX 3080 Ti
conda create -n tensorflow_env python=3.8
pip install tensorflow-gpu=2.5
conda install cudatoolkit cudnn

<파이토치>
가상환경 생성 뒤, 가상환경에서 

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

import torch
torch.cuda.is_available()
반응형

+ Recent posts