딥러닝
tensorflow 2.0 설치
카멜레온개발자
2020. 2. 5. 19:14
현재 tensorflow 2.1까지 나온 상태이고, tutorial 따라해 볼려고 2.0을 설치하려고 함.
* Visual Studio와 cuda, cudnn은 설치되어 있음
(tensorflow 2.0 -> cuda 10.0, tensorflow 2.1 -> cuda 10.1 이 필요한 것 같음)
1. conda create -n tf2_gpu python tensorflow-gpu==2.0
2. 엔터를 쳐서 설치를 진행한다.
3. 파이썬 실행 후 텐서플로우 버전을 확인한다
* 나의 경우 h5py관련 오류가 난다면
(h5py is running against HDF5 1.10.5 when it was built against 1.10.4, this may cause problems......)
다음의 명령을 통해 재설치하면 된다.
pip install h5py --upgrade --no-dependencies --force