Last active
July 10, 2020 04:09
-
-
Save jetsonhacks/acf63b993b44e1fb9528 to your computer and use it in GitHub Desktop.
Install Caffe on NVIDIA Jetson TK1 LT4 21.2 - CUDA 6.5
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/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 |
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.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I maybe get the key. To make the master of caffe for tk1, we should comment compute_60 in makefile.config .just easy