반응형
tensorflow를 사용하던 중에 발생한 failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected 에러가 나는 경우 해결하는 방법입니다.
2024. 3. 14 최초작성
다음 코드를 추가하면 해결됩니다.
| import os os.environ['CUDA_VISIBLE_DEVICES'] = "0" |
참고
https://stackoverflow.com/a/51767213
반응형
'Deep Learning & Machine Learning > Tensorflow&PyTorch 개발환경' 카테고리의 다른 글
| WSL2에서 CUDA 사용할 수 있도록 PyTorch 설치하는 방법 (0) | 2024.03.16 |
|---|---|
| CUDA 버전별 설치가능한 Tensorflow, PyTorch 버전 확인하는 방법 (0) | 2024.03.14 |
| Ubuntu 22.04에 CUDA 사용하도록 Tensorflow 설치하는 방법 (8) | 2024.03.14 |
| Ubuntu / Windows / WSL2 / Apple Silicon Macbook에서 Tensorflow / Pytorch 설치하는 방법 (0) | 2024.03.09 |
| Apple Silicon Macbook에서 PyTorch 설치하기 (0) | 2024.03.09 |
문제 발생시 지나치지 마시고 댓글 남겨주시면 가능한 빨리 답장드립니다.
블로그의 문서는 종종 최신 버전으로 업데이트됩니다. 여유 시간이 날때 진행하는 거라 언제 진행될지는 알 수 없습니다.
영화,책, 생각등을 올리는 블로그도 운영하고 있습니다. https://freewriting2024.tistory.com
제가 쓴 책도 한번 검토해보세요 ^^
