Development/Python
[Linux / Python] 가상환경에 텐서플로 설치
성딱이
2021. 9. 7. 16:48
반응형
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
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
끄읏
반응형