print(torch.cuda.is_available())
# False
- CUDA : 11.2
- cuDNN : 미리 깔았었음
- pytorch : 1.8.1
계속해도 안되서 아래의 페이지를 참고해서
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html위의 command로 torch를 한단계 downgrade했더니 정상작동함.
20.04 우분투기반 NVIDIA GeForce RTX 3090에 CUDA, cuDNN Pytorch 설치
20.04 우분투 기반 NVIDIA GeForce RTX 3090에 CUDA, cuDNN Pytorch 설치 컴퓨터 스팩은 아래와 같으며, 오랜만에 직접 조립 했다. 몇일 써보니 호환이나 큰 문제 없이 돌아가므로 사양정보를 공유 한다. 컴퓨
goodtogreate.tistory.com
'AI' 카테고리의 다른 글
| [Transformer] Positional Encoding (1) | 2024.10.09 |
|---|---|
| RTX 3080 병렬로 tensorflow 위에 올려보기 (0) | 2020.12.23 |
| DQN (Deep Q-Networks) (0) | 2020.10.03 |
| Off-Policy Control (0) | 2020.09.24 |
| RL : RL 용어정리 (0) | 2020.09.08 |
댓글