pytorch2 GPU를 사용한 Pytorch, Tensorflow 설치하기 CUDA, cuDNN을 설치 안 했다면 먼저 아래 링크를 참고하기~ >> GPU 사용을 위한 CUDA, cuDNN 설치하기 - https://ewaterland.tistory.com/19 Pytorch에서 GPU를 쓰는 방법과 Tensorflow에서 GPU를 쓰는 방법이 있다. 둘 다 쓸 거면 둘 다 설치하자 Pytorch 1. CUDA 버전에 맞는 Pytorch 설치하기 - https://pytorch.org/get-started/locally/ PyTorch An open source machine learning framework that accelerates the path from research prototyping to production deployment. pytorch.org Pytor.. 2023. 5. 21. GPU 사용을 위한 CUDA, cuDNN 설치하기 CUDA, cuDNN을 설치 후 Pytorch나 Tensorflow를 깔고 싶다면 밑의 링크를 참고하기~ >> GPU를 사용한 Tensorflow, Pytorch - https://ewaterland.tistory.com/20 1. 자신의 NVIDIA GPU 버전에 맞는 드라이버 설치하기 - https://www.nvidia.com/Download/index.aspx?lang=en-us 검색하면 자동으로 드라이버를 찾아서 추천해준다. 설치하고 나서 재시작해야 함! 터미널에 nvidia-smi 를 입력하면 드라이버가 잘 설치됐는지 알 수 있다. 2. GPU의 CUDA Compute Capability 버전 확인하기 - https://developer.nvidia.com/cuda-gpus - https://.. 2023. 5. 21. 이전 1 다음 반응형
