Skip to content

Instantly share code, notes, and snippets.

@prudhvirajboddu

prudhvirajboddu/install_cuda _for_tensorflow_support.md

Last active January 2, 2021 09:37
Show Gist options
  • Save prudhvirajboddu/0b82f47310f0ad1db79dc3ad7aa6ea1f to your computer and use it in GitHub Desktop.
Save prudhvirajboddu/0b82f47310f0ad1db79dc3ad7aa6ea1f to your computer and use it in GitHub Desktop.
Download ZIP
Installing cuda
Raw
install_cuda _for_tensorflow_support.md

Currently using ubuntu 20.04 on dell g3 machine with 1650Ti Geforce card

Install the nvidia drivers for your PC at https://www.nvidia.com/en-us/drivers/unix/

install the run file from https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu

Deselect the driver installation and continue

Install cudnn8 from https://developer.nvidia.com/rdp/cudnn-download

deb files all installed

GOTO cd /usr/local/cuda/lib64

run sudo cp libcusolver.so.10 libcusolver.so.11 and reboot

Check the installation by installing tensorflow pip install tensorflow for current stable version Goto python interpreter run import tensorflow as tf tf.config.list_physical_devices('GPU')

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment