CUDA

[Linux] Docker container를 이용한 CUDA cuDNN 환경 구축

한 컴퓨터에서 GPU를 사용하는 프로젝트를 하나만 진행한다면 CUDA와 cuDNN이 하나의 버전만 있으면 되겠지만 여러 프로젝트를 진행할 경우, 프로젝트별로 필요한 CUDA, cuDNN 버전이 다를 수 있음 (TensorFlow만 해도 버전별로 각기 다른 CUDA, cuDNN이 필요하다!) 도커를 이용해서 컨테이너별로 다른 버전의 CUDA, cuDNN을 설치해보자 Docker를 이용한 딥러닝 환경 구축 Docker 설치 GPU driver 설치 GPU 드라이버 즉 NVIDIA 드라이버는 base machine에 설치해야함 NVIDIA driver 설치 후 아래의 명령어로 확인 nvidia-smi GPU driver 연결 도커가 GPU를 인식할 수 있도록 NVIDIA Container Toolkit을 이용..