Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
Install Caffe on NVIDIA Jetson TK1 LT4 21.2 - CUDA 6.5
#!/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
@MrThiago

This comment has been minimized.

Copy link

@MrThiago MrThiago commented Apr 15, 2016

Hello Jetsonhacks,
Could you tell us, the performance of it ?
thanks

@zhuhaijun753

This comment has been minimized.

Copy link

@zhuhaijun753 zhuhaijun753 commented Aug 15, 2016

mada some mistakes, because of the version of caffe and cuDNN

@teler

This comment has been minimized.

Copy link

@teler teler commented Oct 14, 2016

Jetson TK1 can use only Cuda 6.5 so it means you can use only cuDNN v2.

@tmkasun

This comment has been minimized.

Copy link

@tmkasun tmkasun commented Nov 20, 2016

There is no branch or tag named dev, since getting following error
error: pathspec 'dev' did not match any file(s) known to git.

@tmkasun

This comment has been minimized.

Copy link

@tmkasun tmkasun commented Nov 20, 2016

Ok finally was able to build the caffe , but not the dev branch, even dev branch does not exist now, They have merged it to master , and master can be only compiled using cuDNN 5+ which require CUDA 7, but currently for Jetson TK1 latest CUDA version is 6.5 and cuDNN last supported version for that is cudnn-6.5-linux-ARMv7-v2.So can't build the master branch in TK1.
Instead, I was able to build this fork which is for cudnn 2.x

@boulderZ

This comment has been minimized.

Copy link

@boulderZ boulderZ commented Nov 22, 2016

I had same error: pathspec 'dev' ... Currently the script is in a random test phase and seems to be passing the tests. What problem am I going to have?

@vntkumar8

This comment has been minimized.

Copy link

@vntkumar8 vntkumar8 commented Jan 7, 2017

able to build using master branch

@halaneji

This comment has been minimized.

Copy link

@halaneji halaneji commented Feb 23, 2017

can you help me to install caffe on ubunto 16.04 please :)

@cuipeng

This comment has been minimized.

Copy link

@cuipeng cuipeng commented Apr 8, 2017

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.

@cuipeng

This comment has been minimized.

Copy link

@cuipeng cuipeng commented Apr 8, 2017

I maybe get the key. To make the master of caffe for tk1, we should comment compute_60 in makefile.config .just easy

@hayday100

This comment has been minimized.

Copy link

@hayday100 hayday100 commented Oct 19, 2017

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.

@MarsMSJ

This comment has been minimized.

Copy link

@MarsMSJ MarsMSJ commented Oct 19, 2017

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment