Skip to content

Instantly share code, notes, and snippets.

@MarsMSJ
Last active August 14, 2018 13:30
Show Gist options
  • Star 3 You must be signed in to star a gist
  • Fork 1 You must be signed in to fork a gist
  • Save MarsMSJ/c97b6b26ba45362a9121dc4fed44c314 to your computer and use it in GitHub Desktop.
Save MarsMSJ/c97b6b26ba45362a9121dc4fed44c314 to your computer and use it in GitHub Desktop.
Jetson TK1 - caffe compilation issue: nvcc fatal : Unsupported gpu architecture 'compute_60'

Jetson TK1 - caffe compilation issue: nvcc fatal : Unsupported gpu architecture 'compute_60'

Summary

I'm posting here a detailed summary of the problem and the situation with the Jetson TK1 board and compiling the final caffe release. The final caffe release requires cudnn v.5. If you attempt to compile on the Jetson TK1 board you will give an error from nvcc saying "Unsupported gpu architecture". I describe the

Problem

On the Jetson TK1 if you clone the final caffe release and compile:
make all -j 4

The nvcc compiler will show the following error:

nvcc fatal : Unsupported gpu architecture 'compute_60'

Solution

wget https://github.com/BVLC/caffe/archive/rc3.tar.gz
tar -xvzf rc3.tar.gz
cd caffe-rc2
make all -j 4

The solution is to download release candidate 2 from Releases. This version will work with the cudnn version (v2) in the NVidia JetPack release for Jetson TK1. The final caffe release requires cudnn v.5 which requires CUDA 9.

Conclusion

Note that Kepler GPUs are supported in the latest CUDA 9 ; only Fermi GPUs are unsupported. The NVIDIA "GK20a" GPU on the Jetson TK1 is a Kepler GPU. Unfortunately, there have been no updates to the Jetson TK1 jetpack in the last release. Downloaded caffe rc2 is the solution for the time being.

@ysh329
Copy link

ysh329 commented Jun 23, 2018

Mark

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