Sign inSign up

pegome/caffe_cuda11

By pegome

•Updated over 3 years ago

Caffe docker images with support of cuda11 and python 3

Image
0

108

pegome/caffe_cuda11 repository overview

⁠caffe_cuda11

Docker image:Caffe, Python3, CUDA 11, Ubuntu 20.04

Docker file can be found at https://github.com/paragpatel/caffe_cuda11⁠

Caffe modified source: https://github.com/paragpatel/caffe/tree/fixes_for_cudnn8_bvlc_master⁠

Tested with GPU RTX 3060

⁠How to use

docker run -it pegome/caffe_cuda11:gpu

Tag summary

Content type

Image

Digest

sha256:3ca5ae3ef…

Size

5.3 GB

Last updated

over 3 years ago

docker pull pegome/caffe_cuda11:gpu