Skip to content

Instantly share code, notes, and snippets.

@jetsonhacks
Last active July 10, 2020 04:09
Show Gist options
  • Save jetsonhacks/acf63b993b44e1fb9528 to your computer and use it in GitHub Desktop.
Save jetsonhacks/acf63b993b44e1fb9528 to your computer and use it in GitHub Desktop.
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
@hayday100
Copy link

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
Copy link

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