Merge pull request #77 from Cyberes/main

Allow hiding of the welcome message

Author: TimDettmers

bitsandbytes/cuda_setup/main.py

@@ -17,6 +17,7 @@
 """
 
 import ctypes
+import os
 
 import torch
 
@@ -112,12 +113,12 @@ def get_compute_capability(cuda):
 
 
 def evaluate_cuda_setup():
-    # we remove this for now and see how things go
-    #print('')
-    #print('='*35 + 'BUG REPORT' + '='*35)
-    #print('Welcome to bitsandbytes. For bug reports, please submit your error trace to: https://github.com/TimDettmers/bitsandbytes/issues')
-    #print('For effortless bug reporting copy-paste your error into this form: https://docs.google.com/forms/d/e/1FAIpQLScPB8emS3Thkp66nvqwmjTEgxp8Y9ufuWTzFyr9kJ5AoI47dQ/viewform?usp=sf_link')
-    #print('='*80)
+    if 'BITSANDBYTES_NOWELCOME' not in os.environ or str(os.environ['BITSANDBYTES_NOWELCOME']) == '0':
+        print('')
+        print('='*35 + 'BUG REPORT' + '='*35)
+        print('Welcome to bitsandbytes. For bug reports, please submit your error trace to: https://github.com/TimDettmers/bitsandbytes/issues')
+        print('For effortless bug reporting copy-paste your error into this form: https://docs.google.com/forms/d/e/1FAIpQLScPB8emS3Thkp66nvqwmjTEgxp8Y9ufuWTzFyr9kJ5AoI47dQ/viewform?usp=sf_link')
+        print('='*80)
     if not torch.cuda.is_available(): return 'libsbitsandbytes_cpu.so', None, None, None, None
 
     cuda_setup = CUDASetup.get_instance()

environment.yml

@@ -1,6 +1,7 @@
 name: 8-bit
 channels:
   - conda-forge
+  - pytorch
 dependencies:
   - python=3.9
   - pytest