-
-
Save jetsonhacks/acf63b993b44e1fb9528 to your computer and use it in GitHub Desktop.
#!/bin/sh | |
# Install and compile Caffe on NVIDIA Jetson TK1 Development Kit | |
sudo add-apt-repository universe | |
sudo apt-get update | |
sudo apt-get install libprotobuf-dev protobuf-compiler gfortran \ | |
libboost-dev cmake libleveldb-dev libsnappy-dev \ | |
libboost-thread-dev libboost-system-dev \ | |
libatlas-base-dev libhdf5-serial-dev libgflags-dev \ | |
libgoogle-glog-dev liblmdb-dev -y | |
sudo usermod -a -G video $USER | |
# Git clone Caffe | |
sudo apt-get install -y git | |
git clone https://github.com/BVLC/caffe.git | |
cd caffe && git checkout dev | |
cp Makefile.config.example Makefile.config | |
make -j 4 all | |
make -j 4 runtest | |
build/tools/caffe time --model=models/bvlc_alexnet/deploy.prototxt --gpu=0 |
can you help me to install caffe on ubunto 16.04 please :)
I had flash tk1 by JetPack first,Then,I run this script, I got some error, the first one is "there is no 'dev' branch " same with tmkasun.
I maybe get the key. To make the master of caffe for tk1, we should comment compute_60 in makefile.config .just easy
This one works for CUDNN v2 on Jetson TK1. https://github.com/platotek/caffetk1. Also make sure to install g++4.8 before you run the make commands.
I just wrote a GIST on this issue.
Jetson TK1 - caffe compilation issue: nvcc fatal : Unsupported gpu architecture 'compute_60'
In short, download caffe rc2 from releases and compile using this build. You need CUDA 9 and yes it sucks that Kepler GPUs are supported just not this one ( Kepler "GK20a" GPU ) and the last Jetpack release had no updates for TK1.
able to build using
master
branch