우선 tensorflow를 설치하기 전에 자신의 환경을 먼저 체크해야합니다.
- CUDA 10.1
- CuDNN 7.6.4
- python 3.7(예정)
저는 tensorflow-gpu를 설치할 것이기 때문에 지금 저희 환경에 맞는것은 1.13 >= 버전입니다.
그렇기 때문에 이번 포스트에서는 최신 버전인 1.14 버전을 설치해보겠습니다.
Linux GPU
Linux
> anaconda(python3.7 / Anaconda 2019.10 for Linux Installer)
https://www.anaconda.com/distribution/#download-section
$ cd Downloads
$ bash Anaconda3-2019.10-Linux-x86_64.sh
yes -> Enter -> yes
$ source ~/.bashrc
> install/remove Virtual Environment
$ conda create -n [virtual environment name] python=3.7 anaconda
$ source activate [virtual environment name] - conda update --all
$ conda remove --name [virtual environment name] --all
$ conda info --envs
> install tensorflow-gpu
# 가상환경 접속 후,
$ conda install tensorflow-gpu=1.14
> install OpenCV
$ conda install -c conda-forge opencv
# 설치확인
$ pkg-config --modversion opencv
$ pkg-config --libs --cflags opencv
만약 다음 error가 난다면
Package opencv was not found in the pkg-config search path.
Perhaps you should add the directory containing `opencv.pc'
to the PKG_CONFIG_PATH environment variable
No package 'opencv' found
$ sudo apt-get install libopencv-dev
> install Pytorch
$ conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
> install etc packages
$ sudo apt-get install git
'OS & Tools > Linux' 카테고리의 다른 글
[Jetson_nano] Install opencv 4.1 (0) | 2020.02.18 |
---|---|
[Jetson_nano] virtualenvwrapper (0) | 2020.02.18 |
[ubuntu 16.04] CUDA & CuDNN 설치 (1) | 2020.02.04 |
[ubuntu 16.04] GTX 1660Ti 드라이버 설치 및 듀얼 모니터 설정 (1) | 2020.02.04 |
[ubuntu 16.04] 나의 데스크탑 사양 및 파티션 설정 (0) | 2020.02.04 |