본문 바로가기
Development/Python

[Linux / Python] 가상환경에 텐서플로 설치

by 성딱이 2021. 9. 7.
반응형

1. 가상환경 활성화

$ conda activate 가상환경명

 

2. 가상환경 활성화 상태 확인(가상환경 내 Python 설치 위치 확인함으로써!)

$ which python

결과 : /data/anaconda3/envs/가상환경/bin/python

별도참조 : https://stackoverflow.com/questions/29980798/where-does-pip-install-its-packages

 

3.1. 텐서플로 설치 (pip사용)

$ pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.5.0-cp38-cp38-manylinux2010_x86_64.whl

참조 : https://www.tensorflow.org/install/pip?hl=ko#1.-install-the-python-development-environment-on-your-system

 

3.2. 텐서플로 설치 (conda 사용)

$ conda search tensorflow 
$ conda install conda install tensorflow=2.4.1=gpu_py38h8a7d6ce_0 -c pkgs/main

(필자의 사양은 TF2.5를 설치해야 하는 상황인데 Conda install에서 설치 할 수 있는 TF2.5-gpu가 없었다. 따라서 1번 방법으로 설치)

 

4. 추가 Dependency 패키지 설치

$ pip install ray==1.5.1

 

끄읏

 

반응형

댓글