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"
]
},
반응형