Bug fixes, shifted installation of gpu libraries only if conda+nvdriver

rsnk96 / Ubuntu-Setup-Scripts

Scripts to help you set up your Ubuntu quickly, especially if you're in any subfield of Data Science or AI!

Apache License 2.0

126 stars 76 forks source link

Closed rsnk96 closed 4 years ago

rsnk96 commented 4 years ago

Fixed detection of cv2 (conda/pip/built from source)
Changed installation of nvtop to be conditional to nvidia driver instead of nvcc
Changed installation of GPU enabled tf and pytorch to only happen if Conda and Display driver are both present, since conda would automatically pick up the right version of conda and cudnn that is required

rsnk96 commented 4 years ago

All changes LGTM