[Python] Jupyter Notebook에서 gpu 사용이 안될 때

cateto·2021년 11월 22일
0

[에러메시지 종류]

CUDA error: no kernel image is available for execution on the device

torch.cuda.is_available()
==> false

참고했던 사이트 : https://stackoverflow.com/questions/60987997/why-torch-cuda-is-available-returns-false-even-after-installing-pytorch-with
https://bluecolorsky.tistory.com/71

여기에 접속해서 자신의 OS 상황에 맞게 command를 얻으면 됨.

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

위와 같았음....
커널 restart하니까 됨

profile
Curious for Everything

0개의 댓글