Tensorflow-gpu 설치 및 Jupyter 등록 on Windows
- Windows, Cuda 9.0, cudnn 7.4, Anaconda는 설치되어 있는 환경에서 진행합니다.
- Cuda 9.0 : https://developer.nvidia.com/cuda-90-download-archive
- Cuda 9.0 설치시 Visual Studio Integration인가 한 메뉴가 있는데 이걸 체크했을때 설치가 잘 안된 기억이 있어서 해제했습니다.
- Cuda쪽만 체크하고 설치해도 무방한듯 합니다.
- cudnn : https://developer.nvidia.com/rdp/cudnn-download
- Cuda 9.0 : https://developer.nvidia.com/cuda-90-download-archive
- Anacomda prompt를 실행합니다.
- 새 가상환경을 만들기 위해 아래를 입력합니다. tf는 가상환경의 이름이고 python version은 3.6으로 했습니다.
conda create -n tf pip python=3.6
- 가상환경을 활성화시킵니다.
activate tf
orsource activate tf
- pip를 업그레이드 시킵니다.
python -m pip install --upgrade pip
- tensorflow-gpu를 설치합니다.
pip install --ignore-installed --upgrade tensorflow-gpu
- Jupyter에 등록하기 위해 아래 명령을 입력합니다.
conda install notebook ipykernel
python -m ipykernel install --user --name tf --display-name "Tensorflow"
- 환경 확인 :
conda info --envs
- conda 환경 삭제 :
conda remove --name ENV_NAME --all
- Jupyter에서 커널 삭제 :
jupyter kernelspec uninstall tf
'머신러닝 > 기타' 카테고리의 다른 글
[PyTorch] PyTorch 설치 및 Jupyter 등록 (0) | 2018.11.11 |
---|