/Ubuntu1804InstallCuda

shell script for installing cuda

Primary LanguageShellMIT LicenseMIT

Ubuntu1804InstallCuda

Step 1

  • Reset environment
  • Install gpu driver

Step 2

  • Install cuda (copied from TF homepage)

Tested 2020/02/23

  • TensorFlow 2.1 works
  • Pytorch 1.4 works