반응형 pytorch1 Nvidia RTX 4060Ti 기반 Ubuntu에서 CUDA + cuDNN + Tensorflow + Pytorch 설치 넉넉한 GPU 메모리 구성으로 가성비 좋은 GeForce RTX 4060Ti 16G를 사용해 Ubuntu 리눅스 환경에서 머신러닝 모델을 연구하고 있습니다. 그동안 TensorFlow와 PyTorch 환경을 셋업하는 과정에서 만난 버전 호환성 문제로 고생을 많이 해서 설치에 성공했던 사례를 글로 정리하여 같은 문제를 겪고 계신 분들에게 도움이 되고자 합니다. GPU 가속 머신러닝 개발환경을 구축하기 위한 첫걸음은 운영체제에서 GPU를 인식하도록 드라이버를 설치하는 것입니다. 그 후에는 TensorFlow와 PyTorch의 명령을 GPU로 전달하기 위해 CUDA와 cuDNN을 설치해야 하며, 마지막으로 TensorFlow 및 PyTorch 를 설치합니다. 각 단계마다 소프트웨어의 버전 호환성이 중요하며, .. 2024. 5. 5. 이전 1 다음 반응형
