pip install --upgrade tensorflow opencv-python matplotlib scipy
-
First install Tensorflow with GPU support here
- WARNING: if you don't have NVIDIA GPU, please skip this part
- NOTE: you can choose driver version yourself, by default I gave 525
- search for the driver versions by running
apt-cache search --names-only "^nvidia-driver-[0-9]{3}$"
sudo apt -y install nvidia-driver-525
sudo reboot
- check the installation by running
nvidia-smi
- WARNING: if you don't have Intel GPU, please skip this part
sudo apt -y install intel-opencl-icd
sudo apt -y install opencl-headers ocl-icd-libopencl1 ocl-icd-opencl-dev clinfo
- check GPU drivers list by running
clinfo -l
(at least you should have one)
pip install --upgrade opencv-python matplotlib scipy
pip install --upgrade opencv-python matplotlib scipy