티스토리 뷰

IT꿀팁

JUPYTER NOTEBOOK에서 CUDA 미인식

tubelog 2019. 11. 20. 20:57

Terminal에서는 CUDA를 정상 인식하지만 Jupyter notebook에서 인식하지 못할 경우,

 

1) .bash_profile 의 $PATH 와 $LD_LIBRARY_PATH에 CUDA 경로가 포함되었는지 확인.

2) Terminal에서 잡혀있으나 notebook에서 ! echo $PATH 로 나오는 결과에는 없을 경우 다음을 실행

3) ! sudo ldconfig 

 

[_DeviceAttributes(/job:localhost/replica:0/task:0/device:CPU:0, CPU, 268435456, 18389685952725290049),
 _DeviceAttributes(/job:localhost/replica:0/task:0/device:XLA_CPU:0, XLA_CPU, 17179869184, 1946781655761808951),
 _DeviceAttributes(/job:localhost/replica:0/task:0/device:XLA_GPU:0, XLA_GPU, 17179869184, 569979717786726748),
 _DeviceAttributes(/job:localhost/replica:0/task:0/device:XLA_GPU:1, XLA_GPU, 17179869184, 10662017307613957212),
 _DeviceAttributes(/job:localhost/replica:0/task:0/device:XLA_GPU:2, XLA_GPU, 17179869184, 5892313992368683549),
 _DeviceAttributes(/job:localhost/replica:0/task:0/device:XLA_GPU:3, XLA_GPU, 17179869184, 6602609572207122371)]

 

그러면 정상적으로 잡힌다. 끝

 

 

무의미한 관련 참고 자료:

https://code-examples.net/ko/q/243fb52

 

python 텐서플로우 파이썬 쉘 내부에서 tensorflow가 GPU 가속을 사용하고 있는지 확인하는 방법은 무엇입니까?

우분투의 내장 apt cuda 설치와 함께 두 번째 대답을 사용하여 우분투 16.04에 tensorflow를 설치했습니다. 이제 내 질문은 tensorflow가 실제로 GPU를 사용하는지 어떻게 테스트 할 수 있습니까? 나는 gtx 960m GPU를 가지고 있습니다. import tensorflow 때 이것이 출력입니다. I tensorflow/stream

code-examples.net

https://tensorflow.blog/6-%EB%B3%91%EB%A0%AC%EC%B2%98%EB%A6%AC-first-contact-with-tensorflow/

 

6. 병렬처리 – First Contact with TensorFlow

이 글은 스페인 카탈루냐 공과대학의 Jordi Torres 교수가 텐서플로우를 소개하는 책 ‘First Contack with TensorFlow’을 번역한 것입니다. 이 글은 원 도서의 라이센스(CC BY-NC-SA 3.0)와 동일한 라이센스를 따릅니다. 목록으로가기 2015년 11월에 처음 나온 텐서플로우 패키지는 GPU를 이…

tensorflow.blog

https://sangeui.tistory.com/112

 

[Ubuntu 18.04] NVIDIA 410 드라이버 및 CUDA 10.0 설치

* https://askubuntu.com/questions/1077061/how-do-i-install-nvidia-and-cuda-drivers-into-ubuntu를 참고하여 진행 * 개인적인 필요로 이를 참고·설치 완료했으며, 해당 글은 타인에게 정보 제공에 있는 것이..

sangeui.tistory.com

 

댓글
댓글쓰기 폼