conda환경에서 cuda version 추가 - jinwooklim/my-exp GitHub Wiki

직접 설치가 아닌, conda 환경에서 cuda를 추가 하고 싶을 때

(conda-env) user@user-All-Series:~$ conda install cudatoolkit

를 사용하자.

pytorch의 경우에 conda가 기본 패키지 관리자이므로 이 방법을 사용하면 대부분 해결 할 수 있다.

하지만, tensorflow의 경우에는 pip가 기본 패키지 관리자이므로 최신 버전과 맞지 않는 등 문제가 있을 수 있다.

아나콘다 설치 후

1. 가상환경 생성

conda create -n 가상환경이름 python=3.6

2. 가상환경 Activation

source activate 가상환경이름

3. TensorFlow GPU 설치

conda install -c anaconda tensorflow-gpu

(2019년 3월 5일 기준 )이렇게 설치하면 현재 아나콘다 패키지에 나와있는 최신 버전인 TensorFlow 1.12 버전으로 설치되며

CUDA 9.2, cuDNN 7.2.1 로 설정된다.

필자는 CUDA 10.0 버전이 설치된 상태에서 가상환경을 위와 같이 설정하였다.

3-1. TensorFlow GPU 버전 설정하여 설치 시

conda install -c anaconda tensorflow-gpu=1.x

conda install -c anaconda cudnn=7.x.x or ...

conda install -c anaconda cudatoolkit=9.x or 10.x ...

(필요시) conda install -c menpo opencv