Sign inSign up

matthewkleinsmith/caffe-pascalgpu

By matthewkleinsmith

•Updated almost 9 years ago

NVIDIA's Pascal GPUs (GTX 10XX) require special Caffe build settings

Image
1

402

matthewkleinsmith/caffe-pascalgpu repository overview

Caffe + NVIDIA's Pascal GPUs requires:

  • CUDA 8.0
  • Building Caffe with the Pascal architecture setting:
  -DCUDA_ARCH_NAME="Manual" \
  -DCUDA_ARCH_BIN="52 60" \
  -DCUDA_ARCH_PTX="60" \

https://github.com/NVIDIA/nvidia-docker/issues/208⁠

Tag summary

Content type

Image

Digest

Size

1.3 GB

Last updated

almost 9 years ago

docker pull matthewkleinsmith/caffe-pascalgpu