본문 바로가기

Deep Learning Tools/Nengo

Nengo Python 환경설정

pip install nengo

pip install nengo_dl

pip install nengo_extra

pip install tensorflow-gpu==1.12.0

pip install numpy

pip install jupyter

pip install matplotlib


nengo-dl은 tensorflow를 1.12 이상 버젼을 쓰기 때문에 주의해야 한다.

이전 버젼 이상은 에러 발생함..


gpu 환경이라면 tensorflow-gpu로 설치, 아닐 경우엔 그냥 tensorflow로 설치


gpu로 설치할 경우, cuda 설치도 해줘야하는데, cuda는 9.0 버젼으로, cuDNN은 7.2 버젼으로 깔아줘야 한다.

역시 버젼 다르면 에러남



1) 기존 cuda 지우기

sudo apt-get --purge remove 'cuda*'
sudo apt-get autoremove --purge 'cuda*'
sudo rm -rf /usr/local/cuda-8.0
sudo rm -rf /usr/local/cuda

2) cuda 설치

apt-get install cuda-9-0

3) cuDNN 설치

apt-get install libcudnn7-dev


4) 설치 확인

cat /usr/local/cuda/version.txt


cat /usr/include/cudnn.h | grep -E "CUDNN_MAJOR|CUDNN_MINOR|CUDNN_PATCHLEVEL"