Deep CV
[PyTorch] Pytorch.cuda.is_available() is false 해결법 본문
CUDA 버전에 따라서 PyTorch 버전을 맞춰줘야 CUDA를 사용할 수 있었다.
$ nvcc -V
Cuda tool의 버전을 확인하고 버전에 맞춰 아래 사이트의 코드로 Pytorch를 install 해줬더니 해결됐다.
PyTorch
An open source machine learning framework that accelerates the path from research prototyping to production deployment.
pytorch.org
옛날 pytorch 버전 설치는 아래 링크
https://pytorch.org/get-started/previous-versions/
++ 추가)
3090, CUDA 11.3 version 사용시 python 3.8은 안 되고, 3.9 버전을 설치하니 해결 됐습니다.
'딥러닝 > PyTorch' 카테고리의 다른 글
| [PyTorch] tensorboard 대충 정리 (0) | 2021.11.02 |
|---|---|
| [PyTorch] .permute() 차원 순서 바꾸기 (0) | 2021.10.29 |
| [Pytorch] torch.clamp (0) | 2021.10.18 |
| [Pytorch] torch.linspace (0) | 2021.10.18 |
'딥러닝/PyTorch' Related Articles
more
