2021. 10. 18. 20:59ㆍ공부한 것 정리/개발환경설정
내 디바이스와 맞지 않는 torch라는 뜻, 무지성으로 pip install requirements.txt하지말고 아래 링크에서 원래 사용하던 torch 버전으로 설치해주자
https://pytorch.org/get-started/previous-versions/
PyTorch
An open source machine learning framework that accelerates the path from research prototyping to production deployment.
pytorch.org
내가 원래 뭘써야하는지 모르겠다 하면 아래 따라하면 됨
1. 내 디바이스의 스펙 알아보고 해당하는 compatibility 확인 (nvidia gpu)
https://en.wikipedia.org/wiki/CUDA
CUDA - Wikipedia
From Wikipedia, the free encyclopedia Jump to navigation Jump to search Parallel computing platform and programming model CUDA (or Compute Unified Device Architecture) is a parallel computing platform and application programming interface (API) that allows
en.wikipedia.org
2. 거기에 맞는 쿠다 버전 설치
3. 쿠다버전에 맞는 cudnn설치
4. 그 버전에 맞는 torch 설치( 위 torch 사이트에서 명령어 따라치기 )
- conda 가상환경 사용한다는 가정하에!!