Install python3-caffe-cuda by entering the following commands in the terminal:
sudo apt update sudo apt install python3-caffe-cuda
Description:
Python3 interface of Caffe (CUDA)
Caffe is a deep learning framework made with expression, speed, and modularity in mind. It is developed by the Berkeley AI Research Lab (BAIR) and community contributors. . It contains the CUDA version of python3 Caffe interface.
Homepage: http://caffe.berkeleyvision.org
Version: 1.0.0-6build1
Section: multiverse/python