Sunday, April 19, 2020

Install Cuda 10.1 & TensorFlow 2.1 on ubuntu 18.04

1. Install Cuda 10.1 & TensorFlow 2.1

@reference_1_tensorflow.org

(1)  Install conda: `conda create -n your_env_name python=3.7 pip`
(2) `pip install tensorflow`

@reference_2_itfashion4u

2. Downgrade versions if necessary

`apt-get install -y --no-install-recommends libnvinfer6=6.0.1-1+cuda10.1 libnvinfer-dev=6.0.1-1+cuda10.1 libnvinfer-plugin6=6.0.1-1+cuda10.1 --allow-downgrades`

3. Uncheck Nvidia repositories to prevent ubuntu software updater from updating 
"Software & Updates" => "Other Software" => uncheck Nvidia repositories