Ubuntu18.04 GPU 버전 Tensorflow2.0+CUDA10.0/Tensorflow2.1+CUDA10.1 가상 환경 구성
13908 단어 Ubuntu&GitTensorFlowKeras
주의: 만약 당신이 이중 시스템이고 드라이버를 설치하지 않았다면 다음 강좌에 따라 드라이버를 설치하는 것을 권장합니다.드라이버가 설치되어 있고 정상적으로 작동할 수 있다면 건너뛰십시오.듀얼 시스템은 시스템 카드 작동을 방지하고 그래픽 드라이브를 장착합니다.
다음 내용 참조: How to install TensorFlow 2.0 on Ubuntu
## 1.종속 라이브러리 1.1 업데이트 시스템 설치
sudo apt-get update
sudo apt-get upgrade
1.2 컴파일 도구 설치
sudo apt-get install build-essential cmake unzip pkg-config
sudo apt-get install gcc-6 g++-6
1.3 screen을 설치하고 같은 창에 여러 개의 터미널 도구를 사용하여 원격 SSH 연결에 사용할 수 있습니다.
sudo apt-get install screen
1.4 install X windows libraries and OpenGL libraries:
sudo apt-get install libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev
1.5 Along with image and video I/O libraries:
sudo apt-get install libjpeg-dev libpng-dev libtiff-dev
sudo apt-get install libavcodec-dev libavformat-dev libswscale-dev libv4l-dev
sudo apt-get install libxvidcore-dev libx264-dev
1.6 install optimization libraries:
sudo apt-get install libopenblas-dev libatlas-base-dev liblapack-dev gfortran
1.7 HDF5 for working with large datasets:
sudo apt-get install libhdf5-serial-dev
1.8 Python 3 development libraries including TK and GTK GUI support:
sudo apt-get install python3-dev python3-tk python-imaging-tk
sudo apt-get install libgtk-3-dev
2 그래픽 드라이브 및 CUDA 설치
비디오 드라이버가 이미 설치되어 있는 경우 전면 2.1~2.5는 건너뛸 수 있습니다.2.1 add an apt-get repository so that we can install NVIDIA GPU drivers.
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
2.2 install your NVIDIA graphics driver:
sudo apt-get install nvidia-driver-418
2.3 reboot command and wait for your system to restart:
sudo reboot now
2.4 both download and install CUDA 10.0 right from your terminal
cd ~
mkdir installers
cd installers/
wget https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux
mv cuda_10.0.130_410.48_linux cuda_10.0.130_410.48_linux.run
chmod +x cuda_10.0.130_410.48_linux.run
sudo ./cuda_10.0.130_410.48_linux.run --override
2.5 error process You will be prompted to accept the End User License Agreement (EULA). During the process, you may encounter the following error:
Please make sure that
PATH includes /usr/local/cuda-10.0/bin
LD_LIBRARY_PATH includes /usr/local/cuda-10.0/lib64, or, add /usr/local/cuda-10.0/lib64 to /etc/ld.so.conf and run ldconfig as root
To uninstall the CUDA Toolkit, run the uninstall script in /usr/local/cuda-10.0/bin
Please see CUDA_Installation_Guide_Linux.pdf in /usr/local/cuda-10.0/doc/pdf for detailed information on setting up CUDA.
*WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 384.00 is required for CUDA 10.0 functionality to work.
To install the driver using this installer, run the following command, replacing with the name of this run file:
sudo .run -silent -driver
Logfile is /tmp/cuda_install_25774.log
You may safely ignore this error message. 2.6 update bash profile
nano ~/.bashrc
Insert the following lines at the bottom of the profile:
# NVIDIA CUDA Toolkit
export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64
여기서 사용자 고유의 CUDA 버전을 수정합니다.
source ~/.bashrc
2.7 query CUDA to ensure that it is successfully installed:
nvcc -V
2.8 cuDNN Library for Linux cuDNN v7.6.4 for CUDA 10.0 from the following link: https://developer.nvidia.com/rdp/cudnn-archive
버전 일치 문제 주의
scp ~/Downloads/cudnn-10.0-linux-x64-v7.6.4.24.tgz \
username@your_ip_address:~/installers
cd ~/installers
tar -zxf cudnn-10.0-linux-x64-v7.6.4.38.tgz
cd cuda
sudo cp -P lib64/* /usr/local/cuda/lib64/
sudo cp -P include/* /usr/local/cuda/include/
cd ~
3. 가상 환경 설치
3.1 download pip3
wget https://bootstrap.pypa.io/get-pip.py
sudo python3 get-pip.py
3.2 install virtual environment tools:
pip3 install virtualenv virtualenvwrapper
3.3 update bash profile
nano ~/.bashrc
insert the following lines at the end of the file:
# virtualenv and virtualenvwrapper
export WORKON_HOME=$HOME/.virtualenvs
export VIRTUALENVWRAPPER_PYTHON=/usr/bin/python3
source /usr/local/bin/virtualenvwrapper.sh
여기가 틀릴 수도 있습니다.virtualenvwrapper 때문입니다.sh 경로 설치가 잘못되었습니다. 오류 신고가 있으면 참고하십시오:/usr/local/bin/virtualen vwrapper.sh: 그 파일이나 디렉터리의 해결 방법이 없습니다
source ~/.bashrc
3.4 create Python 3 deep learning virtual environment named dl4cv:
mkvirtualenv dl4cv -p python3
4. Install TensorFlow 2.0 into your dl4cv virtual environment
CUDA10.1, tensorflow-gpu2.1
4.1 activate the environment and install
workon dl4cv
pip install numpy
pip install tensorflow-gpu==2.0.0 # or tensorflow-gpu==2.1.0
4.2 nstalling standard image processing libraries including OpenCV:
pip install opencv-contrib-python
pip install scikit-image
pip install pillow
pip install imutils
4.3 install machine learning libraries and support libraries
pip install scikit-learn
pip install matplotlib
pip install progressbar2
pip install beautifulsoup4
pip install pandas
4.4 test
workon dl4cv
python
>>> import tensorflow as tf
>>> tf.__version__
2.0.0
>>> import tensorflow.keras
>>> import cv2
>>> cv2.__version__
4.1.2
4.5 check if TensorFlow 2.0’s installation is able to take advantage of your GPU:
workon dl4cv
python
>>> import tensorflow as tf
>>> tf.test.is_gpu_available()
True
4.6 deactivate the current virtual environment:
deactivate
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
PowerShell 기반 Ubuntu 시스템 사용 상세 정보본고는 주로 Ubuntu 16.04 LTS에 PowerShell을 설치하고 사용하는 방법을 소개한다.PowerShell Core는 마이크로소프트가 내놓은 크로스 플랫폼(Windows, Linux, macOS) 자동화...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.