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