AI
Pytorch + RTX3080
월곡동로봇팔
2021. 3. 8. 21:00
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했더니 정상작동함.