Tensorflow, Pytorch 설치 시 유용한 자료

jswboseok·2023년 2월 8일
0

cuda, cudnn, tensorflow 버전 정보

https://www.tensorflow.org/install/source?hl=ko#linux


위와 같이 다운받아야 할 cuda, cudnn, tensorflow 버전을 보여준다.

자신의 cuda, os 환경에 맞는 pytorch 다운로드 방법

https://pytorch.org/get-started/locally/

위에서 자신의 환경에 맞는 command를 밑에 보여준다. 본인은 linux terminal에서 command를 실행하였다.

cuda, cudnn 설치

https://tes-b.github.io/ubuntu/tensorflow-ubuntu_2/
본인은 위 블로그를 참고하였다. 이 때, 환경설정 시 cuda 버전에 맞추어서 command를 입력하도록 하자.

profile
냠냠

0개의 댓글