Programming/Python

Tensorflow 설치

빠릿베짱이 2017. 4. 4. 00:56
반응형

1. 아나콘다 설치 ( 파이썬 3.5 버전)

2. CUDA Toolkit 8.0 설치 ( https://developer.nvidia.com/cuda-downloads )

3. cuDNN v5.1 설치  ( https://developer.nvidia.com/rdp/cudnn-download )

4. GPU 카드가 CUDA Compute Capability 3.0 이상인지 확인 한다. ( https://developer.nvidia.com/cuda-gpus )

cmd 창에서

가상 환경 생성하고

conda create -n tensorflow python=3.5

생성한 가상환경 활성화

activate tensorflow

cpu 버전 설치 시

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.0.1-cp35-cp35m-win_amd64.whl 


gpu 버전 설치 시 

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.0.1-cp35-cp35m-win_amd64.whl 




VSCode에서 python 코드 작성 시

가상 환경의 python으로 실행해야 함으로 python 경로를 생성한 가상환경 경로로 변경한다 ( launch.json 파일 수정 )

{
            "name""Python",
            "type""python",
            "request""launch",
            "stopOnEntry"true,
            "pythonPath""C:/Users/sijoo/Anaconda2/envs/tensorflow/python",
            "program""${file}",
            "cwd""${workspaceRoot}",
            "debugOptions": [
                "WaitOnAbnormalExit",
                "WaitOnNormalExit",
                "RedirectOutput"
            ]
        },




반응형