GPU (2) 썸네일형 리스트형 tensorflow-gpu==1.8.0 을 설치하여도 GPU가 아닌 CPU로 계산할 때, GPU로 계산하도록 하는 방법 tensorflow를 설치하면 cpu 연산을, tensorflow-gpu를 설치하면 gpu연산을 할 수 있다. 따라서 다음 명령어를 이용해서 tensorflow-gpu를 설치한다. pip install --upgrade tensorflow-gpu 위의 명령어를 치면 GPU 연산이 가능해지나 tensorflow-gpu의 최신버전(현재는 tensorflow-gpu==1.13.0) 이 설치된다. 따라서 코드가 실행되도록 하기 위해 tensorflow-gpu를 downgrade 시킨다. pip install tensorflow-gpu==1.8.0 만약 tensorflow-gpu를 설치해도 gpu연산이 되지 않으면 NVIDIA 드라이버가 지워진 것일 수도 있다. 컴퓨터를 재부팅해보고 다음 명령어를 terminal.. [GPU] tensorflow gpu 선택하기 config = tf.ConfigProto() config.gpu_options.allow_growth = False [True 일 경우 필요할 때마다 GPU 사용률이 조절되고 False 는 GPU의 모든 메모리를 점유한다] config.gpu_options.visible_device_list = "1" [GPU 가 2개일 경우, 첫 번째 GPU만 사용하려면 "0", 두번째 GPU만 사용하려면 "1" 로 설정한다.] global_init = tf.global_variables_initializer() local_init = tf.local_variables_initializer() sess = tf.Session(config=config) sess.run([global_init, local_init]) 이전 1 다음