반응형
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 > 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://toss.me/momo2024
제가 쓴 책도 한번 검토해보세요 ^^
