Anaconda 가상환경 GPU 딥러닝 환경 설치(CUDA, CUDNN설치)

https://yjs-program.tistory.com/1

Anaconda설치 Ubuntu 16.04

 

가상환경은 이미 설치했다는 가정하에 GPU 딥러닝 환경에 필수적인 CUDA, CUDNN을 설치해보자.

내가 로컬 환경이 존재한다면, 로컬에 다음 포스팅을 참고해서 그대로 설치해도 괜찮다.

Ubuntu CUDA 설치 + CUDNN

 

다만, 내가 돌리고 있는 코드에 따라 요구되는 환경이 천차만별인데 예를 들어 A프로젝트는 cuda10.1에 cudnn 7.6을 요구하는 환경이고, B프로젝트는 cuda 8.0에 cudnn 5.1을 요구하는 환경이라 가정하면 로컬에 cuda 10, 8을 모두 설치하고 bashrc파일 내 경로를 일일히 변경해가며 설치해줘도 동작하긴 한다.

 

다만, 너무 비효율적이기 때문에 conda 환경에 다음과 같이 설치해주는 것도 한 방법이 될 수 있다.

 

1. 가상환경 활성화

conda activate [$ENV_NAME]

 

2. CUDA, CUDNN 수동 설치

conda install -c anaconda cudatoolkit==[원하는 버전]
ex) conda install -c anaconda cudatoolkit==10.1.243

# CUDA 버전을 못찾는 경우 -c 뒤에 "conda-forge" 옵션 추가
-c conda-forge

# CUDA 먼저 설치하면 아래 커맨드로 어느 정도 상호호환되는 cudnn이 알아서 설치되긴한다.
conda install -c anaconda cudnn

 

3. CUDA, CUDNN 확인

conda list

아래 이미지처럼 CUDNN과 CUDA 환경이 잘 잡힌것을 확인하면 정상적으로 동작할 것이다.