colab 중심으로 해보려고 하다가, GPU환경이 항상확보되는게 아니라 비용지불이 필요해서, 로컬에 환경을 구축해보려고 한다.

 

여기를 보고 하나씩 따라하며 환경 구축 진행 중

 

https://developer.nvidia.com/cuda-gpus 여기에 내 GPU인 1660 super가 나오지는 않지만 구글링 결과 7.5랑 호환 되는거 같다.

여기서 쿠다툴킷 11버전 다운로드 받았다

 

결과적으로 tensorflow는 gpu버전으로 잘 안됐고 pytorch는 됐다.(필요했던건 pytorch라 OK)

 

bash실행이 필요했는데, 윈도우라 안된건 windows10에 포함된 ubuntu를 설치해서 해결했다.

 

반응형

'AI, ML > ML' 카테고리의 다른 글

그레디언트 부스팅 (Gradient Boosting)  (0) 2023.10.15
랜덤 포레스트(random forest)  (1) 2023.10.15
케라스(Keras)  (0) 2020.04.13
Bayesian Online Changepoint Detection 논문리딩  (0) 2019.08.21
부스팅(boosting)  (0) 2019.05.28

+ Recent posts