Skip to content

Instantly share code, notes, and snippets.

View younis125's full-sized avatar

Younis younis125

  • Hohai University
  • China
View GitHub Profile
#!/bin/bash
## This gist contains instructions about cuda v10.1 and cudnn 7.6 installation in Ubuntu 18.04 for Tensorflow 2.1.0
### steps ####
# verify the system has a cuda-capable gpu
# download and install the nvidia cuda toolkit and cudnn
# setup environmental variables
# verify the installation
###
# Install Cuda 10.1 and Cudnn 7.6.5 on Ubuntu 18.04
## Start clean
sudo apt purge nvidia*
sudo apt remove nvidia-*
sudo rm /etc/apt/sources.list.d/cuda*
sudo apt autoremove && apt autoclean
sudo rm -rf /usr/local/cuda*