1. 아나콘다 설치
https://www.anaconda.com/products/individual
2. 텐서플로우, Pytorch 버전 호환 확인
https://www.tensorflow.org/install/source
https://pytorch.org/get-started/previous-versions/
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()
반응형
'머신러닝_딥러닝 > 머신러닝 & 딥러닝' 카테고리의 다른 글
딥러닝 개발환경 구축 (윈도우 기준) (0) | 2022.02.16 |
---|---|
우분투, 딥러닝 환경 구축 1탄 (그래픽카드 & 인터넷 충돌) (0) | 2022.02.04 |
mAP(mean Average Precision) (0) | 2021.12.01 |
휴리스틱(Heuristics)이란? (0) | 2021.10.24 |
퍼지이론 (Fuzzy theory) (0) | 2021.10.24 |